【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 >> 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