-
np.stack的用法
np.stack((a,b,c,...),axis=0)
np.stack 第一個參數(shù)為一個元組
第二個參數(shù)是維度
方法合并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]])