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

Numpy create boolean array from checking if an array index equal a letter

我有一个 numpy 字母数组 'x''y'。我如何为此创建一个布尔数组,如果索引 == 'x' 则 returns 为真,否则为假?

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)