Numpy通过检查数组索引是否等于字母来创建布尔数组

时间:2015-03-17 01:15:33

标签: arrays numpy boolean-logic

我有一大堆字母'x''y'。如何为此创建一个布尔数组,如果索引== 'x'则返回true,否则返回false

1 个答案:

答案 0 :(得分:2)

In [1]: import numpy as np

In [2]: a = np.array(['x', 'y', 'y', 'x', 'y'])

In [3]: a == 'x'
Out[3]: array([ True, False, False,  True, False], dtype=bool)