python - 如何展平在第二个轴上具有不同长度的 2d numpy 数组？

Question

我有一个像这样的 numpy 数组:

myArray = np.array([[1,2],[3]])

但我不能把它弄平，

In: myArray.flatten()
Out: array([[1, 2], [3]], dtype=object)

如果我在第二个轴上将数组更改为相同的长度，那么我可以将其展平。

In: myArray2 = np.array([[1,2],[3,4]])
In: myArray2.flatten()
Out: array([1, 2, 3, 4])

我的问题是:

我可以使用类似 myArray.flatten() 的东西，不管数组的维度和它的元素的长度，并得到输出:array([1,2,3 ])?

Answer 1

myArray 是 对象 的一维数组。您的列表对象将简单地保持与 flatten() 或 ravel() 相同的顺序。您可以使用 hstack水平顺序堆叠数组:

>>> np.hstack(myArray)
array([1, 2, 3])

请注意，这基本上等同于使用 concatenate轴为 1(这应该很直观):

>>> np.concatenate(myArray, axis=1)
array([1, 2, 3])

如果您没有但是可以合并这些项目，最好使用flatten()或ravel() 性能:

In [1]: u = timeit.Timer('np.hstack(np.array([[1,2],[3,4]]))'\
   ....: , setup = 'import numpy as np')
In [2]: print u.timeit()
11.0124390125

In [3]: u = timeit.Timer('np.array([[1,2],[3,4]]).flatten()'\
   ....: , setup = 'import numpy as np')
In [4]: print u.timeit()
3.05757689476

Iluengo's answer您还可以了解有关为什么在给定数组类型的情况下不能使用 flatten() 或 ravel() 的更多信息。