np.delete详解

一、函数

np.delete(array,obj,axis)

二、函数的意思

array:需要处理的矩阵

obj:需要处理的位置，比如要删除的第一行或者第一行和第二行

axis:

如果输入为None：array会先按行展开，然后按照obj，删除第obj-1(从0开始)位置的数，返回一个行矩阵。

如果输入为0：按行删除

如果输入为1：按列删除

三、例子

1、输入为None

import numpy as np
a = np.array([[1,2],[3,4],[5,6]])
a_1 = np.delete(a,2,None) #按行展开后，删除第2 +1个数，因为是从0开始的
print(a_1)

结果

         

 2、输入为0

（1）删除一行

import numpy as np
a = np.array([[1,2],[3,4],[5,6]])
a_1 = np.delete(a,0,0)  #删除第一行
print(a_1)

结果

         

 （2）删除多行

import numpy as np
a = np.array([[1,2],[3,4],[5,6]])
a_1 = np.delete(a,[0,1],0)
print(a_1)

结果

         

 3、输入为1

（1）删除1列

import numpy as np
a = np.array([[1,2,7],[3,4,8],[5,6,9]])
a_1 = np.delete(a,0,1) #删除第一列
print(a_1)

结果

           

（2）删除多列

import numpy as np
a = np.array([[1,2,7],[3,4,8],[5,6,9]])
a_1 = np.delete(a,[0,2],1)
print(a_1)

结果