【Numpy】根据条件批量修改元素的值（numpy.where）

---

引言

Numpy 数组如果想根据条件批量修改元素的值可以使用 numpy.where() 函数。

描述

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

第一参数 condition 是条件，条件成立，返回 x 作为元素的值；否则，返回 y 作为元素值。

示例

单条件

>>> import numpy as np
>>> a = np.arange(10)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> np.where(a < 5, a, 10*a)
array([ 0,  1,  2,  3,  4, 50, 60, 70, 80, 90])

多条件

>>> import numpy as np
>>> a = np.arange(10)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> np.where((a > 3) & (a < 7), a, 10*a)
array([ 0, 10, 20, 30,  4,  5,  6, 70, 80, 90])

参考

https://numpy.org/doc/stable/reference/generated/numpy.where.html