提取熊猫中特定列名称的值

时间:2019-12-27 10:34:37

标签: python json pandas dataframe

假设我有数据框:

    aiman       air    ajang    akhir    aktif     alat
0    0           0       0     0.0196    0.0117   0.0115
1   0.0297    0.0318   0.0223     0        0         0

并且我想将其转换为json文件,并且不想采用值为0的列,例如:

[
    {
     akhir:0.0196,
     aktif:0.0117,
     alat:0.0115
    },
    {
     aiman:0.0297,
     air: 0.0318,
     ajang: 0.0223
    }
] 

我希望我的问题足够清楚。任何人都有一个想法如何获取该json文件?

1 个答案:

答案 0 :(得分:1)

将值转换为词典列表,然后使用0删除值:

d = [{k:v for k, v in x.items() if v != 0} for x in df.to_dict('r')]
print (d)
[{'akhir': 0.0196, 'aktif': 0.0117, 'alat': 0.0115}, 
 {'aiman': 0.0297, 'air': 0.0318, 'ajang': 0.0223}]