python - 如何在numpy中执行重复填充？

Question

我有可变长度的数据，想通过重复较短的样本将其打包成最大样本 len 大小的批处理。

例如从这个

[[0, 1, 2, 3, 4], [0, 1, 2], [2, 2, 3]]

做这个

[[0, 1, 2, 3, 4], [0, 1, 2, 0, 1], [2, 2, 3, 2, 2]]

Answer 1

不是 numpy 答案，但您可以使用 itertools 做到这一点:

from itertools import cycle, islice

lst = [[0, 1, 2, 3, 4], [0, 1, 2], [2, 2, 3]]

n = max(len(item) for item in lst)
res = list(list(islice(cycle(item), n)) for item in lst)
print(res)  # [[0, 1, 2, 3, 4], [0, 1, 2, 0, 1], [2, 2, 3, 2, 2]]

我在哪里使用 cycle循环遍历子列表和 islice获取第一个 n 元素。