如何在转换为数据帧时将字典键保留为第一列?

时间:2017-04-09 16:15:03

标签: python dictionary dataframe

我想将字典转换为数据帧。

dic={'a': [5, 9],
     'b': [4, 61],
     'c': [24, 9]}

如果你这样做:

df=pd.DataFrame(dic).transpose()

您将拥有:

    0   1
a   5   9
b   4  61
c  24   9

但是我想把字典键作为第一列,我该怎么办?

name   F0   F1
a      5    9
b      4    61
c      24   9

1 个答案:

答案 0 :(得分:1)

我认为最简单的是reset_index

df=pd.DataFrame(dic, index=['F0','F1']).rename_axis('name', axis=1).transpose().reset_index()
print (df)
  name  F0  F1
0    a   5   9
1    b   4  61
2    c  24   9

还有:

df=pd.DataFrame.from_dict(dic, orient='index').reset_index()
df.columns = ['name','F0','F1']
print (df)
  name  F0  F1
0    b   4  61
1    c  24   9
2    a   5   9

另一个有点黑客 - 从dict comprehension中的键值添加值,但默认单调索引(0,1,2 ..)也需要reset_index

dic = {k:[k]+v for k,v in dic.items()}
print (dic)
{'b': ['b', 4, 61], 'c': ['c', 24, 9], 'a': ['a', 5, 9]}

df=pd.DataFrame(dic, index=['name','F0','F1']).transpose().reset_index(drop=True)
print (df)

  name  F0  F1
0    a   5   9
1    b   4  61
2    c  24   9