如果整个单元格由特定字符串组成，请将其替换为字符串

Question

我有一个数据帧df:

name     rank
A    captain, general, soldier
B    general, foo, major
C    foo
D    captain, major
E    foo, foo, foo

我想检查列rank中是否有任何单元格由foo组成，以及是否有用foo替换整个单元格。

预期输出：

name     rank
A    captain, general, soldier
B    foo
C    foo
D    captain, major
E    foo

我该怎么做？

Answer 1

df['rank'].replace('.*foo.*', 'foo', regex=True, inplace=True)
# OR
df['rank'].mask(df['rank'].str.contains('foo'), 'foo', inplace=True)
# OR
df.loc[df['rank'].str.contains('foo'), 'rank'] = 'foo'

输出：

name                       rank
0    A  captain, general, soldier
1    B                        foo
2    C                        foo
3    D             captain, major
4    E                        foo

Answer 2

您可以将lambda函数apply添加到列：

df["rank"] = df["rank"].apply(lambda x: "foo" if "foo" in x.split(", ") else x)

在分隔符上进行拆分可以检查单词。例如，世界"；foobar"；不会在其行上触发转换。

编辑：感谢BeRT2me建议用"，"分隔。

Answer 3

mask = df['rank'].str.contains('foo')
df.loc[mask, 'rank'] = 'foo'

Answer 4

if df['rank'].str.contains('foo').any():
df['rank']='foo'