Numpy笔记

where函数

语法

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

condition是一个数组

x和y是可选的

例子

1. 只使用condition参数

import numpy as np

# 创建一个NumPy数组
arr = np.array([10, 20, 30, 40, 50])

# 使用np.where函数找出数组中大于25的元素的位置
indices = np.where(arr > 25)

print(indices)  # 输出: (array([2, 3, 4]),)

2. 加上x和y的参数，小于等于25的元素为-1

# 使用np.where函数找出数组中大于25的元素
result = np.where(arr > 25, arr, -1)

print(result)  # 输出: array([-1, -1, 30, 40, 50])