温馨提示×

numpy怎么删除指定元素

小亿
100
2024-01-31 14:32:40
栏目: 编程语言

使用NumPy库中的delete()函数可以删除指定的元素。delete()函数的用法如下:

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

其中,arr是要操作的数组,obj是要删除的元素的索引或切片对象,axis是要删除的轴。如果axis未指定,则将数组展开为一维数组,然后删除指定的元素。

下面是一个示例,演示了如何使用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]

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
new_arr = np.delete(arr, 1, axis=0)  # 删除第1行
print(new_arr)
# 输出:
# [[1 2 3]
#  [7 8 9]]

new_arr = np.delete(arr, 1, axis=1)  # 删除第1列
print(new_arr)
# 输出:
# [[1 3]
#  [4 6]
#  [7 9]]

在以上示例中,np.delete()函数分别删除了一维数组和二维数组中的指定元素。

0