Я пытаюсь использовать сравнение массивов 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'])