检查系列的哪些元素是给定文本的子字符串 - Python、Pandas

Question

我想做的是：

给定一个字符串序列，找到字符串的所有索引， 它们是另一个主字符串的子字符串，以矢量化方式。

输入：

series = pd.Series(['ab', 'abcd', 'bcc', 'abc'], name='text')
main_text = 'abcX'

# The series:
0      ab
1    abcd
2     bcc
3     abc
Name: text, dtype: object

所需的输出：

0      ab
3     abc
Name: text, dtype: object

我尝试过的：

df_test = pd.DataFrame(series)
df_test['text2'] = main_text
df_test['text'].isin(df_test)

# And this of course won't work, since it check if the main string is a 
# substring of the series strings:
series.str.contains(main_text, regex=True)

谢谢！

Answer 1

您不需要正则表达式，只需使用 in:

series[[e in main_text for e in series]]

输出：

0     ab
3    abc
Name: text, dtype: object