前言
np.hstack的功能 —— 数组元素的堆叠
一、栗子
import numpy as np
arr1 = [1, 2, 3]
arr2 = [4, 5]
arr3 = [6, 7]
res = np.hstack((arr1, arr2,arr3))
print(res) # [1 2 3 4 5 6 7]
import numpy as np
arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])
res = np.hstack((arr1, arr2))
print(res)
结果为:
[[1 2 5 6]
[3 4 7 8]]
二、总结
使用np.hstack()可方便进行数组堆叠