如何从 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])。
3,4,7
a
index=[2,3,6]
您可以使用函数从 NumPy 数组中删除特定元素np.delete,该函数允许您指定要删除的元素的索引。以下是使用示例执行此操作的方法:
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保持不变。