python中多个数组的行合并以及列合并

最近遇到数组合并问题,以此记录解决方法,供大家参考学习。

import numpy as np

a = [1, 2, 3, 4]
b = [5, 6, 7, 8]

#数组纵向合并
mergeA = np.vstack((a, b))
print("mergeA: ", mergeA)
#mergeA:  [[1 2 3 4]
#          [5 6 7 8]]

#数组横向合并
mergeB = np.hstack((a, b))
print("mergeB: ", mergeB)
#mergeB:  [1 2 3 4 5 6 7 8]

参考链接
删除操作

# 单行操作
import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]]) 
print(np.delete(a, 1, axis=0)) # 行删
print(np.delete(a, -1, axis=1)) # 列删
 
输出:
[[1 2 3]
 [7 8 9]]
[[1 2]
 [4 5]
 [7 8]]

# 多行操作
# 注意,当删除多行或多列时时,不能使用倒序下标-1,-2
import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]]) 
print(np.delete(a, [0,1], axis=0)) # 行
print(np.delete(a, [1,2], axis=1)) # 列
 
输出:
[[7 8 9]]
[[1]
 [4]
 [7]]

你可能感兴趣的:(python)