我有以下数据结构:
import pandas as pd
import json
df = pd.DataFrame( {'g1' : ['ABC', 'ABC', 'XYZ', 'XYZ'], 'g2' : ['DEF', 'GHI', 'RST', 'UVW']})
print df
>> g1 g2
0 ABC DEF
1 ABC GHI
2 XYZ RST
3 XYZ UVW
我正在尝试编写具有以下结构的 JSON 文件
$ cat ABC.json
> {
"DEF" : true,
"GHI" : true
}
和
$ cat XYZ.json
> {
"RST" : true,
"UVW" : true
}
到目前为止,我已经能够创建聚合
print df.groupby(u'g1',as_index=True)[u'g2'].aggregate(lambda x: set(x))
>> g1
ABC {GHI, DEF}
XYZ {RST, UVW}
并将其转储到 JSON
dd = json.loads(df.to_json())
然后将其写入单个文件
for k,v in dd.iteritems():
with open(k+'json','wb') as fp:
json.dump(v, fp)
但是以pandas-y的方式将列表映射到dict仍然让我感到困惑。我将发布我的 python(非熊猫)答案作为参考
倚天杖
慕村9548890
相关分类