我有numpy字符串数组,我想针对引用字符串进行测试,并根据字符串数组的每个元素是否包含引用字符串输出布尔数组。我有一个解决方案,但希望有更优雅/高效的解决方案,可能在纯python中实现。感谢您的任何意见。
import numpy as np
import pandas as pd
import re
myarray = np.array(['abc1', 'abc2', 'abc3'])
refstring = 'c2'
pd.Series(myarray).apply(lambda x: re.search(refstring, x)).astype('bool')
>> 0 False
>> 1 True
>> 2 False
>> dtype: bool
答案 0 :(得分:1)
根据我的评论:
map(lambda x: refstring in x, ['abc1', 'abc2'...])