python-3.x - 如何为数据框的一列转换某些值？

Question

假设有一个数据框 df

A     B

0   Aus Brazilian Restaurant

1   Aus Gym

2   Aus Zoo

3   Aus Restaurant Thai

4   Aus Australian restaurant

需要将其转换为

A     B

0   Aus Restaurant

1   Aus Gym

2   Aus Zoo

3   Aus Restaurant

4   Aus Restaurant

创建数据框的代码：

temp = {'A':5*['Aus'],'B':['Brazilian Restaurant','Gym','Zoo','Restaurant Thai','Australian restaurant']
     }

temp

df = pd.DataFrame(temp)

df

这是Ipython中的代码

标签： python-3.xdataframe

Answer 1

我找到了解决方案。在这里张贴，以防它对某人有用。

for i in range(df.shape[0]):
if ('Restaurant' in df.loc[i,'B'] or 'restaurant' in df.loc[i,'B']):
    df.loc[i,'B']='Restaurant'

df