删除numpy数组的某一行或者某一列

函数

numpy.delete(arr, obj, axis=None)

arr：要对哪个数组进行操作

axis：axis=0对行操作，axis=1对列操作

obj：删除obj行/列，如果obj是数组，则代表删除多行/多列

例子

a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
a
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

a_0_0=np.delete(a, obj=0, axis=0)
a_0_0 
array([[4, 5, 6],
       [7, 8, 9]])

a_01_0=np.delete(a, obj=[0, 1], axis=0)
a_01_0
array([[7, 8, 9]])


a_0_1=np.delete(a, obj=0, axis=1)
a_0_1
array([[2, 3],
       [5, 6],
       [8, 9]])

a_01_1=np.delete(a, obj=[0, 1], axis=1)
a_01_1
array([[3],
       [6],
       [9]])

参考

numpy 删除一列_numpy 删除第一个-CSDN博客

 (7)numpy.delete删除np数组的某一行或某一列_delete清空某一列-CSDN博客