-
np.stack的用法
np.stack((a,b,c,...),axis=0)
np.stack 第一个参数为一个元组
第二个参数是维度
方法合并2个np.array，并增加一个维度
In [2]: import numpy as np
In [3]: a = np.array([1,2])
In [4]: b = np.array([3,4])
In [5]: c = np.stack((a,b),axis=0)
In [6]: d = np.stack((a,b),axis=1)
In [7]: c
Out[7]:
array([[1, 2],
[3, 4]])
In [8]: d
Out[8]:
array([[1, 3],
[2, 4]])
