Выбор элементов в массиве numpy с использованием регулярных выражений

Можно выбрать элементы в numpy массивах следующим образом

a = np.random.rand(100)
sel = a > 0.5 #select elements that are greater than 0.5
a[sel] = 0 #do something with the selection

b = np.array(list('abc abc abc'))
b[b==a] = 'A' #convert all the a's to A's

Это свойство используется np. где функция для получения индексов:

indices = np.where(a>0.9)

Я хотел бы иметь возможность использовать регулярные выражения при таком выборе элементов. Например, если я хочу выбрать элементы из b выше этого соответствовать регулярному выражению [Aab] , мне нужно написать следующий код:

regexp = '[Ab]'
selection = np.array([bool(re.search(regexp, element)) for element in b])

Это выглядит слишком многословным для меня. Есть ли более короткий и элегантный способ сделать это?

9
задан Boris Gorelik 6 July 2011 в 11:39
поделиться