numpy连接数组

numpy学习专题

十三、numpy连接数组

连接数组

concatenate() 
原型：numpy.concatenate((a1,a2,...),axis)
作用：用于沿指定轴连接相同形状的两个或多个数组
参数                      说明
a1,a2,...             相同类型的数组
axis                  沿着它连接数组的轴，默认为0

import numpy as np
arr = np.arange(1,9).reshape(2,4)
arr2 = np.arange(11,19).reshape(2,4)
print(arr,"\n")
print(arr2,"\n")
print(np.concatenate((arr,arr2)),"\n")

print(np.concatenate((arr,arr2),1),"\n")

[[1 2 3 4]
 [5 6 7 8]] 

[[11 12 13 14]
 [15 16 17 18]] 

[[ 1  2  3  4]
 [ 5  6  7  8]
 [11 12 13 14]
 [15 16 17 18]] 

[[ 1  2  3  4 11 12 13 14]
 [ 5  6  7  8 15 16 17 18]] 

stack()

原型：numpy.stack(arrays,axis)
作用：用于沿新轴连接数组序列
参数                       说明
arrays                    相同形状的数组序列
axis                      数组中的轴，输入数组沿着它来堆叠

import numpy as np
arr = np.arange(1,5).reshape(1,4)
arr2 = np.arange(11,15).reshape(1,4)
print(arr,"\n")
print(arr2,"\n")
print(np.stack((arr,arr2)),"\n")
print(np.stack((arr,arr2)).shape,"\n")
print(np.stack((arr,arr2)).ndim,"\n")


print(np.stack((arr,arr2),1),"\n")
print(np.stack((arr,arr2),1).shape,"\n")
print(np.stack((arr,arr2),1).ndim,"\n")

[[1 2 3 4]] 

[[11 12 13 14]] 

[[[ 1  2  3  4]]

 [[11 12 13 14]]] 

(2, 1, 4) 

3 

[[[ 1  2  3  4]
  [11 12 13 14]]] 

(1, 2, 4) 

3 

hstack()

是numpy.stack函数的变体，它通过水平堆叠来生成数组

import numpy as np
arr = np.arange(1,9).reshape(2,4)
arr2 = np.arange(11,19).reshape(2,4)
print(np.hstack((arr,arr2)))

[[ 1  2  3  4 11 12 13 14]
 [ 5  6  7  8 15 16 17 18]]

vstack()

是numpy.stack函数的变体，它通过竖直堆叠来生成数组

import numpy as np
arr = np.arange(1,9).reshape(2,4)
arr2 = np.arange(11,19).reshape(2,4)
print(np.vstack((arr,arr2)))

[[ 1  2  3  4]
 [ 5  6  7  8]
 [11 12 13 14]
 [15 16 17 18]]

一定要支持作者哦