python - 在python中用numpy展平数组的正确有效方法？

Question

这个问题在这里已经有了答案:

From ND to 1D arrays (7 个回答)

关闭 6年前。

我有:

a = array([[1,2,3],[4,5,6]])

我想将其展平，将两个内部列表连接到一个平面数组条目中。我能做到:

array(list(flatten(a)))

但由于列表转换，这似乎效率低下(我想以数组而不是生成器结束。)

另外，如何将其推广到这样的数组:

b = array([[[1,2,3],[4,5,6]], [[10,11,12],[13,14,15]]])

结果应该在哪里:

b = array([[1,2,3,4,5,6],
           [10,11,12,13,14,15]])

是否有内置/高效的 numpy/scipy 操作符？谢谢。

Answer 1

您可能需要查看 numpy.flatten 和 numpy.ravel，它们都从 n 维数组返回一维数组。

此外，如果您不打算修改返回的一维数组，我建议您使用 numpy.ravel，因为它不会复制数组，而只是返回数组的 View ，比 numpy.flatten 快得多。

>>>a = np.arange(10000).reshape((100,100))

>>>%timeit a.flatten()
100000 loops, best of 3: 4.02 µs per loop

>>>%timeit a.ravel()
1000000 loops, best of 3: 412 ns per loop

还可以查看 post .