首页 > 综合 > 科技资讯 >

📚 NumPy `where()` 函数用法详解 🌟

发布时间:2025-04-01 00:21:24来源:

NumPy 是 Python 中处理数组的强大库,而 `where()` 函数则是其中非常实用的一个工具。简单来说,`where()` 的作用是根据条件筛选数据,堪称数组操作的“智能筛选器”。🤔

首先,让我们看看它的基本语法:

```python

numpy.where(condition, [x, y])

```

- condition:定义筛选条件,返回布尔值数组。

- x, y(可选):如果条件为 True,则返回 x;否则返回 y。

例如:

```python

import numpy as np

arr = np.array([1, 2, 3, 4, 5])

result = np.where(arr > 3, '大于3', '小于等于3')

print(result)

输出:['小于等于3' '小于等于3' '小于等于3' '大于3' '大于3']

```

不仅如此,`where()` 还能用于多维数组的复杂操作,比如定位特定元素或批量修改数据。💡

例如:

```python

arr = np.array([[10, 20], [30, 40]])

result = np.where(arr > 25, arr 2, arr)

print(result)

输出:[[ 1040] [ 6080]]

```

无论是简化代码逻辑,还是提升运行效率,`where()` 都是 NumPy 用户的必备技能!💪✨

免责声明:本答案或内容为用户上传,不代表本网观点。其原创性以及文中陈述文字和内容未经本站证实,对本文以及其中全部或者部分内容、文字的真实性、完整性、及时性本站不作任何保证或承诺,请读者仅作参考,并请自行核实相关内容。 如遇侵权请及时联系本站删除。