【发布时间】:2021-12-10 08:38:49
【问题描述】:
我遇到了这个问题,请帮忙:
我有一个数据框和一个系列如下:
import pandas as pd
df1 = pd.DataFrame({"ID": [4,8,35,28,34,34,14,28],
"cause of failure": [5,3,0,1,7,8,6,7],
"crash": [0,0,0,1,1,1,1,1]})
我有一个系列:
s = pd.DataFrame({'':["None","Design flaw","metal fatigue","Manufacturing flaw","Pilot
error","Mechanical failure","Improper maintenance","Fire","Corrosion"]})
#要添加新列,我编写了 add_column 函数,在函数内部我将 s1 转换为 #a 字典,但是
def add_column(df,s, ref_column, new_column):
dict_s = s.to_dict(orient='dict')
#dict_s = s.loc[0:].to_dict(orient='index')
error', 5: 'Mechanical failure', 6: 'Improper maintenance', 7: 'Fire', 8: 'Corrosion'}
df[new_column] = df[ref_column].map(dict_s)
return df
add_column(df1,s1,"failure code","failure")
#我不希望空头进入字典,只希望序列的键值对
{'': {0: 'None', 1: 'Design flaw', 2: 'metal fatigue', 3: 'Manufacturing flaw', 4: 'Pilot error', 5: 'Mechanical failure', 6: 'Improper maintenance', 7: 'Fire', 8: 'Corrosion'}}
#Like this:
{0: 'None', 1: 'Design flaw', 2: 'metal fatigue', 3: 'Manufacturing flaw', 4: 'Pilot error', 5: 'Mechanical failure', 6: 'Improper maintenance', 7: 'Fire', 8: 'Corrosion'}
我找不到字典
#然后我想映射这个字典以匹配“故障代码”列中的代码并将它们添加为新列“故障” 最后,我想向 df1 数据框添加一个新的列故障。
结果应该如下:
ID failure code crash failure
1 8 3 0 Manufacturing flaw
2 35 0 0 None
3 28 1 1 Design flaw
4 34 7 1 Fire
5 34 8 1 Corrosion
6 14 6 1 Improper maintenance
7 28 7 1 Fire
【问题讨论】:
标签: python dataframe dictionary series