如何通过比较字符串来创建数组掩码？

Question

如果值是某个值，我尝试使用numpy的数组比较来创建掩码。

虽然可以轻松比较数组是否包含数字值，但该方法不适用于字符串数组。

num_array = np.array([2, 2, 3])
string_array = np.array(['a', 'a', 'b'])

num_array == 2 # returns array(['True', 'True', 'False'])
string_array == 'a' # returns False

的预期结果

string_array == 'a' # array(['True', 'True', 'False'])