python - 在 pandas 中查找子字符串

Question

我有一个数据，其中某些行中有单词。例如:

测试字符串

(测试1)字符串

测试(字符串1)

我需要使用 pandas 查找括号中的子字符串。因此，此处的输出将为 ['Test1', 'String1']

我尝试了类似的方法，但我找不到括号中的确切单词。

df['column'].str.extract('([A-Z]\w{0,})')

Answer 1

您可以使用以下正则表达式模式:

In [180]:
df['text'].str.extract(r'\((\w+)\)')

Out[180]:
0        NaN
1      Test1
2    String1
Name: text, dtype: object

所以这会查找括号中出现的任何单词，这里括号需要转义 \( 例如，我们还想找到所有单词，所以 w+ 是这里需要。

如果您想要一个列表，可以调用dropna，然后调用tolist:

In [185]:
df['text'].str.extract(r'\((\w+)\)').dropna().tolist()

Out[185]:
['Test1', 'String1']