python-3.x - 逐行堆叠 numpy 数组

Question

我正在尝试按行堆叠两个 numpy 数组。我可以这样做。有没有更好的办法？

>>> a = np.arange(7, 13).reshape(2, 3)
>>> b = np.arange(1, 7).reshape(2, 3)
>>> x = np.row_stack((a[0],b[0]))
>>> y = np.row_stack((a[1],b[1]))
>>> z = np.stack((x, y))
>>> z
array([[[ 7,  8,  9],
        [ 1,  2,  3]],

       [[10, 11, 12],
        [ 4,  5,  6]]])

Answer 1

你可以得到同样的:

import numpy as np
a = np.arange(7, 13).reshape(2, 3)
b = np.arange(1, 7).reshape(2, 3)
z = np.stack((a, b), axis=1)
print(z)
# [[[ 7  8  9]
#   [ 1  2  3]]
# 
#  [[10 11 12]
#   [ 4  5  6]]]