熊猫-根据另一列(城市名称)创建一个新列(分支名称)

时间:2020-08-25 07:11:49

标签: python pandas dataframe city

我有以下Python Pandas Dataframe(8行):

City Name
     
New York
Long Beach
Jamestown
Chicago
Forrest Park
Berwyn
Las Vegas
Miami

我想根据城市名称添加一个新的列(分支名称),如下所示:

City Name       Branch Name

New York        New York
Long Beach      New York
Jamestown       New York
Chicago         Chicago
Forrest Park    Chicago
Berwyn          Chicago
Las Vegas       Las Vegas
Miami           Miami 

我该怎么做?

1 个答案:

答案 0 :(得分:0)

您可以使用.map()。不在词典中的城市名称将被保留。

df["Branch Name"] = df["City Name"].map({"Long Beach":"New York",
                                         "Jamestown":"New York",
                                         "Forrest Park":"Chicago",
                                         "Berwyn":"Chicago",}, na_action='ignore')
df["Branch Name"] = df["Branch Name"].fillna(df["City Name"])