np.flatten():合并两个数组ndarray

今天遇到要把两个np.ndarray合并在一起的情况:

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

先叠一起:

c = np.vstack([a,b])
print(c)
array([[1, 1, 2, 3],
       [1, 1, 2, 3]])

然后就可以用flatten()合并了:

c.flatten()
array([1, 1, 2, 3, 1, 1, 2, 3])

参数order不一样的时候,合并的方式也不一样

c.flatten(order=1)
array([1, 2, 1, 3, 2, 4, 3, 5])#可以看到是按着列和并的

你可能感兴趣的:(numpy)