我的 Pandas DataFrame 中有一個名為“State”的列。它包含美國州的縮寫。我有硬編碼的地區(qū)，我想為每個州的地區(qū)創(chuàng)建一個新列。我使用了 pd.Series.apply()，但我想知道這種類型的映射是否有更好的做法。關(guān)于如何改進我的代碼的任何建議？這是我當(dāng)前有效的代碼，但我只是愿意就最佳實踐提出建議。def get_region(s, *regions): if s in regions[0]: return 'west' elif s in regions[1]: return 'midwest' elif s in regions[2]: return 'south' elif s in regions[3]: return 'northeast' else: return Nonewest = ['WA','OR','CA','ID','NV','MT','WY','UT','AZ','CO','NM']midwest = ['ND','MN','WI','MI','SD','NE','KS','IA','MO','IL','IN','OH']south = ['TX','OK','AR','LA','MS','TN','KY','AL','GA','FL','SC','NC','VA','WV','MD','DE']northeast = ['PA','NJ','NY','CT','MA','RI','VT','NH','ME']regions = [west,midwest,south,northeast]full_df['Region'] = full_df['State'].apply(get_region, args=regions)full_df['Region'].head(15)Out:0 west1 midwest2 south3 south4 midwest5 west6 south7 south8 west9 midwest10 south11 northeast12 northeast13 west14 westName: Region, dtype: object
2 回答
侃侃爾雅
TA貢獻1801條經(jīng)驗 獲得超16個贊
檢查 map
s=pd.DataFrame([west,midwest,south,northeast],index=['west','midwest','south','northeast'])
s=s.reset_index().melt('index')
full_df['Region'] = full_df['State'].map(dict(zip(s['value'],s['index'])))
慕桂英4014372
TA貢獻1871條經(jīng)驗 獲得超13個贊
您可以嘗試創(chuàng)建一個 dict 并將其映射到列：
west_dict = {i:"west" for i in west}
midwest_dict = {i:"midwest" for i in midwest}
south_dict = {i:"south" for i in south}
northeast_dict = {i:"northeast" for i in northeast}
d = {**west_dict, **midwest_dict, **south_dict, **northeast_dict}
full_df['Region'] = full_df['State'].map(d)
添加回答
舉報
0/150
提交
取消