python - 将 Pandas 数据框转换为字典，其中键是索引，值是列值列表

Question

我有一个数据框

df:

    cola    colb   colc   cold
0      0    'a'     'b'   'c'
1      1    'd'     None  None
2      2    'g'     'h'   None

我想将它转换成 dict，其中索引是键，列值列表是如下所示的值:

d = {0 : [0,'a','b','c'], 1: [1,'d'], 2: [2,'g','h'] }

我尝试过的:

df.to_dict(orient='index')

我也尝试在 orient 参数中使用其他值，但没有任何效果。

编辑:

我想忽略输出中显示的字典中的 NULL 值。

Answer 1

使用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']}