np.concatenate((a1, a2, ...), axis=0)
concatenate函数沿着某axis轴将a1、a2等数组进行合并。
总是记不住axis对应的方向,看到官网的examples觉得很容易懂,记录一下。
python
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6]])
np.concatenate((a, b), axis=0)
array([[1, 2],
[3, 4],
[5, 6]])
np.concatenate((a, b.T), axis=1)
array([[1, 2, 5],
[3, 4, 6]])
np.concatenate((a, b), axis=None)
array([1, 2, 3, 4, 5, 6])
其中axis(读着aeksis,记住这里的x发音ks就可以)为0和1时分别代表不同的拼接方向。
当axis为0时,数组沿着横轴拼接,这就要确保拼接的数组a和b有相同维度的列数。
当axis为1时,数组沿着纵轴拼接,这里就要确保a和b有相同维度的行数。