如何转换数据框的某一列的某些值?

时间:2019-02-16 17:37:17

标签: python-3.x dataframe

说有一个数据帧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中的代码

1 个答案:

答案 0 :(得分: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