NumPy：不同大小的 concat/stack ndarrays

Question

我有多个不同大小的 ndarray（例如 a = np.arange(3)、b = np.arange(4)，我想将它们连接/堆叠到一个 ndarray res 与 dtype ndarray：

> res
array([array([0, 1, 2]), array([0, 1, 2, 3])], dtype=np.ndarray)

有没有办法做到这一点？

Answer 1

考虑到答案Merge two numpy array's of different shape into a single array，您可以使用 numpy 模块中的函数 concatenate：

a = np.arange(3)
b = np.arange(4)
c = np.concatenate((a, b))