numpy矩阵拼接常用方法:
支持数组和数组或数组和数的拼接,不支持三个及以上数组的拼接
参数:
注:
1、如果axis没有给出,则默认axis=None,arr,values都将先展平成一维数组。
2、如果axis被指定了,那么arr和values需要同为一维数组或者有相同的shape,否则报错:ValueError: arrays must have same number of dimensions
3、axis的最大值为数组arr的维数-1,如arr维数等于1,axis最大值为0;arr维数等于2,axis最大值为1,以此类推。
4、当arr的维数为2(理解为单通道图),axis=0表示沿着行增长方向添加 values;axis=1表示沿着列增长方向添加values
5、当arr的维数为3(理解为多通道图),axis=0,axis=1时同上;axis=2表示沿着图像深度增长方向添加values
import numpy as np
a = [[1,2,3,4]]
b = 5
c = [[5,6,7,8]]
d = np.append(a,b) # 数组和数拼接,默认axis=None
e = np.append(a,c) # 数组和数组拼接,默认axis=None
f = np.append(a,c,axis=0) # 按行增长方向拼接(垂直拼接)
g = np.append(a,c,axis=1) # 按列增长方向拼接(水平拼接)
print(d)
print(e)
print(f)
print(g)
功能与np.append()类似,但是支持多个数组的拼接。
参数:
import numpy as np
a = [[1,2,3,4]]
b = 5
c = [[5,6,7,8]]
d = np.concatenate((a,b),axis=None) # 数组和数展平成一维数组拼接
e = np.concatenate((a,c)) # 数组和数组拼接,默认axis=0,按行增长方向拼接(垂直拼接)
f = np.concatenate((a,c),axis=1) # 数组和数组拼接,按列增长方向拼接(水平拼接)
print(d)
print(e)
print(f)
同样支持多矩阵拼接,不同的是,stack会在指定轴方向上添加一个新的维度,axis默认值为0
参数:
import numpy as np
a = np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12]])
b = np.array([[13,14,15,16],[17,18,19,20],[21,22,23,24]])
print(a.shape)
print(b.shape)
c = np.stack((a,b)) # 默认axis=0,数组和数组在0轴拼接,并在该纬度增加一维
d = np.stack((a,b),axis=1) # axis=1,数组和数组在1轴拼接,并在该纬度增加一维
e = np.stack((a,b),axis=2) # axis=2,数组和数组在2轴拼接,并在该纬度增加一维
print(c, c.shape)
print(d, d.shape)
print(e, e.shape)
水平堆叠,对多维数组来说,水平堆叠相当于在第二个维度做concatenation
import numpy as np
a = np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12]])
b = np.array([[13,14,15,16],[17,18,19,20],[21,22,23,24]])
c = np.hstack((a,b))
print(a, a.shape)
print(b, b.shape)
print(c, c.shape)
垂直堆叠,对多维数组来说,垂直堆叠相当于在第一个维度做concatenation
import numpy as np
a = np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12]])
b = np.array([[13,14,15,16],[17,18,19,20],[21,22,23,24]])
c = np.vstack((a,b))
print(a, a.shape)
print(b, b.shape)
print(c, c.shape)