当前位置:首页python > 正文

Python如何删除numpy数组中指定值的元素

作者:野牛程序员:2023-08-11 09:55:36python阅读 2902

在 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
new_arr = np.delete(arr, np.where(arr == value_to_remove))
print(new_arr)

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

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


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

最新推荐

热门点击