Python中的numpy.delete

NumPy是Python中常用的用于科学计算的库。其中一个功能是numpy.delete，可以用于从一个数组中删除指定的元素或子数组。

基本用法

import numpy as np

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

new_arr = np.delete(arr, 2)  # 删除下标为2的元素

print(new_arr)  # 输出[1 2 4 5]

numpy.delete的第一个参数是要删除元素的数组，第二个参数是要删除的元素的下标。在上述例子中，我们删除了下标为2的元素，即数字3。

在二维和多维数组中的用法

import numpy as np

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

new_arr = np.delete(arr, 1, axis=0)  # 删除第二行

print(new_arr)
# 输出[[1 2]
#      [5 6]]

在二维数组中，我们可以使用numpy.delete删除指定的行或列。如果要删除行，可以设置axis为0；如果要删除列，可以设置axis为1。在上述例子中，我们删除了第二行，即[3, 4]。

import numpy as np

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

new_arr = np.delete(arr, 0, axis=1)  # 删除第一列

print(new_arr)
# 输出[[2]
#      [4]
#      [6]]

如果要删除列，可以设置axis为1。在上述例子中，我们删除了第一列，即[1, 3, 5]。

import numpy as np

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

new_arr = np.delete(arr, 1, axis=0)  # 删除第二个子数组

print(new_arr)
# 输出[[[1 2]
#       [3 4]]]

在多维数组中，我们可以使用numpy.delete删除指定的子数组。在上述例子中，我们删除了第二个子数组，即[[5, 6], [7, 8]]。

结论

numpy.delete是一个方便的函数，可以用于从数组中删除指定的元素或子数组。通过设置axis参数，还可以用于多维数组的操作。