python - python 将二维数组合并到列表中

Question

我想做的是像这样转换一个二维数组:

np.array([[ 0, 1, 2, 3], [ 1, 5, 6, 7]])

进入此(包含所有数字的列表):

[0,1,2,3,1,5,6,7]

有什么办法可以实现吗？

Answer 1

x = np.array([[ 0, 1, 2, 3], [ 1, 5, 6, 7]])    

list(x.flat)    # if you want a list
#  [0, 1, 2, 3, 1, 5, 6, 7]

x.flatten()  # if you want a numpy array
#   array([0, 1, 2, 3, 1, 5, 6, 7])

我不清楚您是否想要一个列表或 numpy 数组，但它们都很容易获得(尽管我假设您想要一个列表，因为您用 list 标记了这个问题)。选择您想要的或对您最有用的选项是合理的。

对于多种用途，numpy has significant advantages over lists ，但有时列表效果更好。例如，在许多构造中，一次获取一个项目，并且提前不知道结果输出数组的大小，在这种情况下，使用 append 构建一个列表是有意义的> 然后将其转换为 numpy 数组以进行 FFT。

还有其他方法可以在列表和 numpy 数组之间进行转换。当您需要做一些不同的事情(例如，更快)时，请务必查看文档或在此处询问。