python的numpy np.where()例子

python的numpy np.where()

说明

np.where() 是numpy中的一个函数，它的作用是根据一个条件返回两个数组中的元素。

语法格式

np.where(condition, x, y)

其中，condition 是一个条件，x 和 y 是两个数组，当condition为True时返回 x 中对应的元素，否则返回 y 中对应的元素。

例子

import numpy as np

x = np.array([1, 2, 3, 4, 5])
y = np.array([10, 20, 30, 40, 50])

z = np.where(x > 3, x, y)

print(z)

输出结果

[10 20 30 4 5]