测试字符串数组的每个元素并输出布尔数组

时间:2017-04-24 18:45:50

标签: python arrays string

我有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

1 个答案:

答案 0 :(得分:1)

根据我的评论:

map(lambda x: refstring in x, ['abc1', 'abc2'...])