У меня есть массив numpy, который выглядит следующим образом:
a_few_lists=
array([['A', 'B', '-', 'C', 'D'],
['E', 'F', 'G', '-', 'H'],
['I', 'J', '-', 'K', 'L']], dtype='<U2')
Я хочу найти в каждой строке '-' и создать массив, который помечает оставшиеся слева от него как true. Я также хочу удалить этот элемент. Я получил индексы для строки, используя np.nonzero(a_few_lists == '-')[1]
, возвращая [2,3,2]
. Я мог бы использовать их для удаления «-» довольно легко, но я также хочу отметить другие элементы.
Желаемый результат:
array([['A', 'B', 'C', 'D'],
['E', 'F', 'G', 'H'],
['I', 'J', 'K', 'L']], dtype='<U2')
array([[True, True, False, False],
[True, True, True, False],
[True, True, False, False]])