当前位置:首页python > 正文

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

作者:野牛程序员:2023-08-11 10:00:22python阅读 2873

在 NumPy 中,要删除数组中特定元素,你可以使用布尔索引或者使用 np.delete() 函数。以下是使用这两种方法的示例:

  1. 使用布尔索引: 使用布尔索引来创建一个不包含特定元素的新数组。

import numpy as np

arr = np.array([1, 2, 3, 4, 3, 5])
value_to_remove = 3
new_arr = arr[arr != value_to_remove]
print(new_arr)
  1. 使用 np.delete() 函数: 使用 np.delete() 函数来删除数组中特定元素。

import numpy as np

arr = np.array([1, 2, 3, 4, 3, 5])
value_to_remove = 3
indices_to_remove = np.where(arr == value_to_remove)[0]
new_arr = np.delete(arr, indices_to_remove)
print(new_arr)

在第二种方法中,np.where(arr == value_to_remove) 返回一个包含要删除元素的索引的数组,然后 np.delete() 函数根据这些索引删除相应的元素。

无论使用哪种方法,这些操作都会生成一个新的数组,原始数组不会受到影响。请注意,在 NumPy 中,由于数组是固定大小的,删除元素会生成一个新的数组而不是修改原始数组。


野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892
野牛程序员教少儿编程与信息学竞赛-微信|电话:15892516892
相关推荐

最新推荐

热门点击