Numpy：获取数组的元素，该元素在比较中包含True

Question

import numpy as np  
import re

list1= ['651ac1', '21581', '13737|14047', '22262', '12281', '12226', '61415', '61495']
regexp = '[a-zA-Z]'
selection = np.array([bool(re.search(regexp, element)) for element in list1])
if True in selection:
    #get_element_containing_true

selection如下所示：

selection
array([ True, False, False, False, False, False, False, False, False], dtype=bool)

我想得到数组中包含True的元素。我怎么得到这个？你知道吗

Answer 1

你真的需要这里的numpy（如果不需要，请参阅@Divakar's answer）？如果这样做，可以将list1转换为np.array并索引：

np.array(list1)[selection]

这被称为^{}。以防你感兴趣。你知道吗

---

性能提示：如果您多次使用正则表达式：编译它并重用已编译的表达式：

regexp = re.compile('[a-zA-Z]')
selection = np.array([bool(regexp.search(element)) for element in list1])

这可以更快更容易地与另一个答案结合起来。你知道吗