一、刪除數(shù)組中的元素

numpy中的np.delete函數(shù)可以用來(lái)刪除數(shù)組中指定位置的元素。它的語(yǔ)法是：

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

舉個(gè)例子，假設(shè)有如下的一個(gè)數(shù)組：

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
print(arr)

array([[1, 2],
       [3, 4],
       [5, 6]])

我們可以使用np.delete來(lái)刪除第一行和最后一行：

new_arr = np.delete(arr, [0, 2], axis=0)
print(new_arr)

array([[3, 4]])

在上面的例子中，我們傳入了一個(gè)列表[0, 2]作為要?jiǎng)h除的行的下標(biāo)，將第一行和最后一行刪除了。

二、刪除數(shù)組中的多個(gè)元素

如果要?jiǎng)h除多個(gè)元素，可以將它們的下標(biāo)放在一個(gè)列表中，然后傳給np.delete函數(shù)。比如下面的例子中我們刪除了數(shù)組中的第1列和第3列：

new_arr = np.delete(arr, [0, 2], axis=1)
print(new_arr)

array([[2],
       [4],
       [6]])

三、刪除數(shù)組中的指定元素

除了按照行或列刪除一整段數(shù)據(jù)，np.delete還可以按照元素刪除。

例如，有如下一個(gè)數(shù)組：

arr = np.array([1, 2, 3, 4, 5])
print(arr)

array([1, 2, 3, 4, 5])

我們可以使用np.delete來(lái)刪除其中的一些元素：

new_arr = np.delete(arr, [0, 2])
print(new_arr)

array([2, 4, 5])

在上面的例子中，我們傳入了一個(gè)列表[0, 2]，將數(shù)組中的第1個(gè)和第3個(gè)元素刪除了。

四、刪除多個(gè)數(shù)組中的元素

如果要?jiǎng)h除多個(gè)數(shù)組中的元素，可以將它們連接在一起，然后一起傳給np.delete函數(shù)。比如下面的例子中我們刪除了數(shù)組a和數(shù)組b中的下標(biāo)為1的元素：

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.concatenate((a, b))
new_c = np.delete(c, 1)

print(new_c)

array([1, 3, 4, 5, 6])

五、刪除元素時(shí)保持?jǐn)?shù)組的維度

有時(shí)候在刪除元素時(shí)需要保持?jǐn)?shù)組的維度，這時(shí)可以給axis參數(shù)傳入一個(gè)元組，來(lái)指定要沿哪些軸方向刪除元素。比如下面的例子，我們刪除掉數(shù)組中的第1行和第3行，但是要保持?jǐn)?shù)組的維度不變：

arr = np.array([[1, 2], [3, 4], [5, 6]])
new_arr = np.delete(arr, (0, 2), axis=(0, 0)).reshape(1, 4)
print(new_arr)

array([[3, 4, 5, 6]])

六、總結(jié)

np.delete函數(shù)可以用來(lái)刪除數(shù)組中指定位置的元素，可以按照行或列刪除一整段數(shù)據(jù)，也可以按照元素刪除。如果要?jiǎng)h除多個(gè)數(shù)組中的元素，可以將它們連接在一起，然后一起傳給np.delete函數(shù)。如果要?jiǎng)h除元素時(shí)需要保持?jǐn)?shù)組的維度，可以給axis參數(shù)傳入一個(gè)元組，來(lái)指定要沿哪些軸方向刪除元素。