小能豆

如何删除 numpy 数组中的特定元素

py

如何从 numpy 数组中删除一些特定元素?假设我有

import numpy as np

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

然后我想3,4,7从中删除a。我只知道值的索引(index=[2,3,6])。


阅读 25

收藏
2024-09-22

共1个答案

小能豆

您可以使用函数从 NumPy 数组中删除特定元素np.delete,该函数允许您指定要删除的元素的索引。以下是使用示例执行此操作的方法:

import numpy as np

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

# Remove elements at the specified indices
result = np.delete(a, indices_to_remove)

print(result)

这将输出:

[1 2 5 6 8 9]

np.delete函数创建一个新数组,并删除指定的元素,因此a保持不变。

2024-09-22