python where

作者：末世修罗发布日期:2025-09-14 浏览:34

import numpy as np

# 创建一个数组
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])

# 使用 np.where 查找满足条件的元素索引
# 这里我们查找所有大于 5 的元素的索引
indices = np.where(arr > 5)

print("满足条件的元素索引:", indices)

# 输出满足条件的元素
print("满足条件的元素:", arr[indices])

解释说明：

np.where(condition)：返回满足条件的元素的索引。condition 是一个布尔表达式，用于指定查找的条件。
在上面的代码中，arr > 5 会生成一个布尔数组 [False, False, False, False, False, True, True, True, True]，表示哪些元素大于 5。
np.where(arr > 5) 返回的是一个元组，包含满足条件的元素的索引。
最后，通过 arr[indices] 可以获取满足条件的元素值。

如果你需要其他类型的 where 用法，请具体说明。

上一篇：python 判断空

下一篇：python 判断数字