np.concatenate函数

np.concatenate((a1, a2, …), axis=0)
concatenate函数沿着某axis轴将a1、a2等数组进行合并。
总是记不住axis对应的方向,看到官网的examples觉得很容易懂,记录一下。

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有相同维度的行数。

你可能感兴趣的:(numpy,python,开发语言)