numpy中where()

np.where()是NumPy库中一个非常有用的函数,它根据给定条件从一个数组中筛选元素。

函数语法:

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

参数解释:

• condition是一个布尔数组或布尔表达式,用于判断对应元素的真值条件。
• x和y是可选参数,需要与condition输出的布尔数组的形状相同。

返回值:

        一个元组,其中第一个元素是condition为True的元素在输入数组中的索引;如果提供了x和y,第二个元素是输出数组,由x和y根据condition进行选择组成。

例子:

import numpy as np

a = np.arange(10, 20)

# 1. 找出大于5的元素
print(np.where(a > 15))
# 输出: (array([6, 7, 8, 9]),)

 注意这里返回的是坐标.也就是说，当where()只传入一个参数时，返回的是符合条件的坐标。

a = np.arange(10)

# 2. 将大于5的元素赋值为10,小于等于5的赋值为a的原值
b = np.where(a > 5, 10, a)
print(b)
# 输出: [ 0  1  2  3  4  5 10 10 10 10]

# 3. 找出非0元素的坐标
c =