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]])