stack(x,axis=0)
这个函数用于将多个数组合并,其中每个数组的shape都相同,其中axis表示在第几个空间开始进行结合,这样说很难理解,举例说明:
[0,1,2,3] 的shape可看做(1,4)
a = np.arange(4)
b = np.arange(4)
c=np.stack((a,b),axis=0)
print(c)
print(c.shape)
输出:
[[0 1 2 3]
[0 1 2 3]]
(2, 4)
c=np.stack((a,b),axis=1)
输出:
[[0 0]
[1 1]
[2 2]
[3 3]]
(4, 2)
当axis等于0时,在0这个维度每个元素包含的内容是[0 1 2 3],所以匹配的内容也就是[0 1 2 3],并且元素只有这一个。
当axis等于1时,则匹配第二个维度,两个数组的第二个维度的元素都分别为1,2,3,4,每次匹配都会产生相应的结果,在这里分别为[0 0],[1 1],[2 2],[3 3]。
再举个例子:
a = np.arange(4).reshape(2,2)
b= np.arange(4).reshape(2,2)
c=np.stack((a,b),axis=0)
print(c)
print(c.shape)
输出: #匹配的元素为数组本身
[[[0 1]
[2 3]]
[[0 1]
[2 3]]]
(2, 2, 2)
c=np.stack((a,b),axis=1)
print(c)
print(c.shape)
输出: #匹配的元素为[0 1],[2 3]
[[[0 1]
[0 1]]
[[2 3]
[2 3]]]
(2, 2, 2)
c=np.stack((a,b),axis=2)
print(c)
print(c.shape)
输出: #匹配的元素为0,1,2,3
[[[0 0]
[1 1]]
[[2 2]
[3 3]]]
(2, 2, 2)
这里的数组的shape为(2,2),但在这个函数中一定要看成(1,2,2)