我有一个要在下面计算的“单词”列表
word_list = ['one','three']
我在pandas数据框中有一个专栏，下面有文字。
TEXT |
-------------------------------------------|
"Perhaps she'll be the one for me." |
"Is it two or one?" |
"Mayhaps it be three afterall..." |
"Three times and it's a charm." |
"One fish, two fish, red fish, blue fish." |
"There's only one cat in the hat." |
"One does not simply code into pandas." |
"Two nights later..." |
"Quoth the Raven... nevermore." |
所需的输出如下所示，其中保留原始文本列，但仅将word_list中的单词提取到新列中
TEXT | EXTRACT
-------------------------------------------|---------------
"Perhaps she'll be the one for me." | one
"Is it two or one?" | one
"Mayhaps it be three afterall..." | three
"Three times and it's a charm." | three
"One fish, two fish, red fish, blue fish." | one
"There's only one cat in the hat." | one
"One does not simply code into pandas." | one
"Two nights later..." |
"Quoth the Raven... nevermore." |
有没有办法在Python 2.7中做到这一点？
用途str.extract：
df['EXTRACT'] = df.TEXT.str.extract('({})'.format('|'.join(word_list)),
flags=re.IGNORECASE, expand=False).str.lower().fillna('')
df['EXTRACT']
0 one
1 one
2 three
3 three
4 one
5 one
6 one
7
8
Name: EXTRACT, dtype: object
输入的每个单词word_list都由正则表达式分隔符连接|，然后传递给str.extract进行正则表达式模式匹配。
re.IGNORECASE为区分大小写的比较而打开了该开关，并且将结果匹配项转换为小写形式以与您的预期输出匹配。
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句