我有一个数据框
数据框:
cola colb colc cold
0 0 'a' 'b' 'c'
1 1 'd' None None
2 2 'g' 'h' None
我想将其转换为
dict
,其中index是键,column值列表是如下所示的值:d = {0 : [0,'a','b','c'], 1: [1,'d'], 2: [2,'g','h'] }
我试过的:
df.to_dict(orient='index')
我还尝试了
orient
参数中的其他值,但没有成功。编辑:
我想忽略字典中的空值,如输出中所示。
最佳答案
与DataFrame.to_dict
一起使用orient='list'
,仅在转置DataFrame
之前:
d = df.T.to_dict(orient='list')
print (d)
{0: [0, 'a', 'b', 'c'], 1: [1, 'd', 'e', 'f'], 2: [2, 'g', 'h', 'i']}
编辑:
d = df.stack().groupby(level=0).apply(list).to_dict()
print (d)
{0: [0, 'a', 'b', 'c'], 1: [1, 'd'], 2: [2, 'g', 'h']}
或:
d = {k:[x for x in v if x is not None] for k, v in df.T.to_dict(orient='list').items()}
print (d)
{0: [0, 'a', 'b', 'c'], 1: [1, 'd'], 2: [2, 'g', 'h']}