Using the np.stack()
function you can specify which axis
would you like to be considered the index axis. So as you can see you will never get a shape of 6
, only (2,3)
or (3,2)
for this example depending on what axis you chose.
See below:
A = np.array([1, 2, 3])
B = np.array([2, 3, 4])
arrays = [A, B]
With this code:
print(np.stack(arrays, axis=0))
you get this output:
[[1 2 3]
[2 3 4]]
with this code:
print(np.stack(arrays, axis=1))
you get this output:
[[1 2]
[2 3]
[3 4]]
与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…