如何展平 dtype 對象的 numpy 數組

Question

我正在使用不同長度的 ndarray 切片，我希望我的結果是平坦的。 例如：

a = np.array(((np.array((1,2)), np.array((1,2,3))), (np.array((1,2)), np.array((1,2,3,4,5,6,7,8)))))

有沒有直接的方法通過使用 numpy 功能（沒有循環）使這個數組變平？

Answer 1

您可以嘗試將其展平，然后使用hstack ，它將數組按順序水平堆疊。

>>> a = np.array(((np.array((1,2)), np.array((1,2,3))), (np.array((1,2)), np.array((1,2,3,4,5,6,7,8)))))
>>> np.hstack(a.flatten())
array([1, 2, 1, 2, 3, 1, 2, 1, 2, 3, 4, 5, 6, 7, 8])