连接两个不同维度的 arrays numpy

Question

我有两个 numpy arrays 形式的 np.array([1,2,3])) 我想连接它们所以我得到：

[[1,2,3],[4,5,6]] 

当存在 [[1,2,3],[4,5,6]] 形式的 numpy 数组和另一个 [7,8,9] 数组时，它也必须工作，我想得到：

[[1,2,3],[4,5,6],[7,8,9]]

我试过 np.concatenate 但我无法让它工作

Answer 1

您可以为此使用vstack ：

import numpy as np

a = np.array([1,2,3])
b = np.array([4,5,6])
c = np.array([7,8,9])

d = np.vstack((a,b))

e = np.vstack((d, c))

print(d)
print(e)

给出：

[[1 2 3]
 [4 5 6]]    

[[1 2 3]
 [4 5 6]
 [7 8 9]]