python - Numpy 数组 python 维度统一

Question

我有一个二维数组，其中一维有 15 个元素，第二维长度可变

例如

>>print abc.size()
15
>>print abc[0].size()
5873
>>print abc[1].size()
9825

如何使用 numpy 或 skikit 稀疏数组使数组维度统一。数据是图像的特征。

Answer 1

假设你想将所有数组向左对齐，并用零填充到右侧，那么你可以首先找到最大长度

max_len = max([abc[i].size() for for i in range(abc.size())])

然后使用 zeros 填充:

import numpy as np

for i in range(abc.size()):
    abc[i] = np.append(abc[i], np.zeros(max_len - abc[i].size())