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)

結果