用python m次复制列表

假设图像在 (n, x, y) (n, x, y, RGB) 成形阵列:

np.repeat(A, B.shape[0] // A.shape[0], axis = 0)

如果你真的想保留列表,我还是建议你在 numpy

list(np.repeat(np.array(A), len(B) // len(A), axis = 0))

A = [a for a in A for _ in range(len(B) // len(A))]
if len(B) % lenA > 0:
    B = B[:-(len(B) % len(A))]

我把这个例子调低了一点,使它有3个图像,并设想每个图像需要复制7次( m==7 ). IIUC,这应该可以:

import numpy as np

def flatten_list_of_lists(replicate):
    for item in replicate:
        if isinstance(item, list):
            yield from flatten_list_of_lists(item)
        else:
            yield item

A = [np.arange(9).reshape(3,3), np.arange(28).reshape(7,4), np.arange(20).reshape(2,10)]
m = 7
replicate = [[x]*7 for x in A]
result = list(flatten_list_of_lists(replicate))
print([x.shape for x in result])

[(3, 3), (3, 3), (3, 3), (3, 3), (3, 3), (3, 3), (3, 3), (7, 4), (7, 4), (7, 4), (7, 4), (7, 4), (7, 4), (7, 4), (2, 10), (2, 10), (2, 10), (2, 10), (2, 10), (2, 10), (2, 10)]

result (列表)