numpy入门——数组组合

数组组合

-合并两个数组 在竖直方向

# 带入numpy包以np的形式import numpy as np# 创建一个一维数组 a1 = np.array([1,2,3])# 创建一个一维数组a2 = np.array([4,5,6])# 合并两个数组  在竖直方向b = np.vstack((a1,a2))print(b)

array([[1, 2, 3],       [4, 5, 6]])

• 合并两个数组 在水平方向

d = np.hstack((a1,a2))print(d)

array([1, 2, 3, 4, 5, 6])

• 给列上的数据增加一个维度

e = a1[:,np.newaxis]print(e)

array([[1],       [2],       [3]])

print(np.hstack((e,e)))print(np.vstack((e,e)))

array([[1, 1],       [2, 2],       [3, 3]])array([[1],       [2],       [3],       [1],       [2],       [3]])

-使用concatenate函数进行更加通用的合并方式

# axis==1在行上进行合并print(np.concatenate((e,e),axis = 1))# axis==0 在列上进行合并np.concatenate((e,e),axis = 0)

array([[1, 1],       [2, 2],       [3, 3]])array([[1],       [2],       [3],       [1],       [2],       [3]])