如果满足条件则替换Numpy元素

>>> import numpy as np>>> a = np.random.randint(0, 5, size=(5, 4))>>> aarray([[4, 2, 1, 1],       [3, 0, 1, 2],       [2, 0, 1, 1],       [4, 0, 2, 3],       [0, 0, 0, 2]])>>> b = a < 3>>> barray([[False,  True,  True,  True],       [False,  True,  True,  True],       [ True,  True,  True,  True],       [False,  True,  True, False],       [ True,  True,  True,  True]], dtype=bool)>>> >>> c = b.astype(int)>>> carray([[0, 1, 1, 1],       [0, 1, 1, 1],       [1, 1, 1, 1],       [0, 1, 1, 0],       [1, 1, 1, 1]])

您可以使用以下方法来缩短它：

>>> c = (a < 3).astype(int)