最近遇到数组合并问题,以此记录解决方法,供大家参考学习。
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]]