1

我想检查 pandas 数据框列是否包含列表中的特定字符串,如果为真,则将该字符串放入新列中。我知道我可以用很多循环来做到这一点,但我认为根据我在文件中得到的行数,它会很慢。

In [0]: mylist = ['ABC', 'ACB', 'DCB', 'AXR12-1']


In [1]: df
Out[1]: 
          text                  
0 ABC KEY778 20225 
1 XXX11 K ABC 14452GXX
2 ACB AZZXG112 
3 DCB AA112G AXXX
4 DCB AA15G AXR12-1

结果

In [2]: df
Out[2]: 
          text            extract     
0 ABC KEY778 20225       ABC
1 XXX11 K ABC 14452GXX   ABC
2 ACB AZZXG112           ACB
3 DCB AA112G AXXX        DCB
4 DCB AA15G AXR          DCB AXR12-1

你有一个简单的方法来做到这一点吗?谢谢

4

0 回答 0