用熊猫替换字符串

Question

我有一个 pandas 列，其中包含一些字符串值，例如：

White bear
Brown Bear
Brown Bear 100 Kg
White bear 200 cm             

如何检查所有字符串是否包含序列“白熊”并将整个值（不仅是序列）替换为像“White_bear”这样的字符串？

df['Species'] = df['Species'].str.replace('White bear', 'White_bear')   

不适合我，因为它只替换了序列。

Answer 1

您可以使用布尔索引：

In [173]: df.loc[df.Species.str.contains(r'\bWhite\s+bear\b'), 'Species'] = 'White_bear'

In [174]: df
Out[174]:
             Species
0         White_bear
1         Brown Bear
2  Brown Bear 100 Kg
3         White_bear

或更通用的解决方案：

In [204]: df
Out[204]:
             Species
0         White bear
1         Brown Bear
2  Brown Bear 100 Kg
3  White bear 200 cm

In [205]: from_re = [r'.*?\bwhite\b\s+\bbear\b.*',r'.*?\bbrown\b\s+\bbear\b.*']

In [206]: to_re = ['White_bear','Brown_bear']

In [207]: df.Species = df.Species.str.lower().replace(from_re, to_re, regex=True)

In [208]: df
Out[208]:
      Species
0  White_bear
1  Brown_bear
2  Brown_bear
3  White_bear

RegEx explanation