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