在Pandas中使用数组值创建JSON密钥

我有一个数据框被操作了几次.最终结果是一个如下所示的数据框：

    Company volume
1   NV      14
2   GD      8
3   AB      6

我将这个数据帧转换为json格式,如下所示：

formattedDF.to_json(orient='columns')

但是,我得到这样的东西：

'{"company":{"1":"NV","2":"GD","3":"AB"},"volume":{"1":14,"2":8,"3":6}}'

这与我想要的非常接近,但我正在寻找的格式是：

'{"company": ["NV", "GD","AB"], "volume":[14, 8, 6}]'

其中键以数组格式给出值.我怎样才能做到这一点？

解决方法:

要么：

print(df.to_dict(orient='list'))

输出：

{'Company': ['NV', 'GD', 'AB'], 'volume': [14, 8, 6]}