追加numpy数组而不展平

时间:2019-04-04 10:47:34

标签: python arrays numpy append

假设我有三个这样的数组:

Total = [], A = [1,2,3]B = [4,5,6]

当我使用Total = np.append(Total,A)时,它会给我:[1,2,3]

当我使用Total = np.append(Total,B)时,它会给我:[1,2,3,4,5,6]

第二次添加后我想要的是:Total = [[1,2,3],[4,5,6]]

2 个答案:

答案 0 :(得分:0)

为什么不将它们堆叠?

A = [1,2,3]
B = [4,5,6]

Total = np.vstack((A,B))    
print(Total)  

输出

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

OR

只需添加列表:

T = [A] + [B]    
print(T)

输出

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

答案 1 :(得分:0)

您可以通过以下几种方法执行此操作:

import numpy as np

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

np.vstack([a, b])
np.stack([a, b], axis=0)
np.c_[a, b].T
np.concatenate([a[np.newaxis], b[np.newaxis]])