Flatten numpy array with sub-arrays of different dimensions

Question

这似乎很简单，但是我还没有找到如何使用来做的事情numpy。考虑示例数组：

import numpy as np
aa = np.array([np.array([13.16]), np.array([1.58 , 1.2]), np.array([13.1]), np.array([1. , 2.6])], dtype=object)

我需要一种通用的方法，使用将该数组展平为单个N元素数组N=every float in all the sub-arrays。在这种情况下，它将是：

aa = np.array([13.16, 1.58 , 1.2, 13.1, 1. , 2.6])

我试过了np.ndarray.flatten()（尝试了所有的“ order”选项），但是我又得到了相同的不变aa数组。

为什么np.ndarray.flatten()不工作，我该怎么做？

解决方案应该尽可能通用，因为aa我在这里使用的示例数组实际上将在我的真实代码中填充不同长度的子数组。

Answer 1

您可以使用 numpy.hstack

>>> np.hstack(aa)
array([13.16,  1.58,  1.2 , 13.1 ,  1.  ,  2.6 ])