python - 如何调整 numpy.array 中的尺寸

Question

全部

我有一个 numpy 数组，元素的维度不同。例如：

[
 [1,2,3],
 [2,3,4,5,6]
 [1,2]
]

我想调整元素尺寸，将尺寸设置为最大，并填充为0，怎么办？

标签： pythonnumpy

Answer 1

您可以先创建具有所需形状的占位符，然后用data列表填充占位符

data = [[1,2,3], [2,3,4,5,6], [1,2]]

# create a placeholder
tmp = np.zeros((len(data), max([len(item) for item in data])))

# fill the placeholder with data
for ind, line in enumerate(data):
    tmp[ind, :len(line)] = line

但是，当数据列表的大小很大时，这可能不是超级快。