Pandas - 将CSV转换为JSON - 使用groupby进行组织，然后删除索引键

Question

我有一个CSV文件，我正在尝试使用Pandas转换为JSON。 它有多个标题但是为了简单起见，我们只说它有3个：“region”，“state”和“salesperson”。 三列，三行通常具有重复值（州名等）。

我理想的结果是：

{
    "salesperson": [
        {
            "name": "John Doe",
            "values": [
                {
                    "region": "North America",
                    "state": "Connecticut"
                },
                {
                    "region": "North America",
                    "state": "Vermont"
                }
            ]
        },
        {
            "name": "Jane Doe",
            "values": [
                {
                    "region": "North America",
                    "state": "New York"
                },
                {
                    "region": "North America",
                    "state": "New Hampshire"
                }
            ]
        }
    ]
}

这就是我目前用于读取数据并将其转换为JSON的方法。

df = pd.read_csv('Foo.csv', encoding="ISO-8859-1",
                    escapechar='\\')
result = (df.groupby(['salesperson'])
            .apply(lambda x: x.to_dict('r'))
            .to_json(orient='table')
            )
return result

.to_json(orient='table')很接近，它给了我

"data": [
    {
        "salesperson": "John Doe",
        "values": [
            {
                "region": "North America",
                "state": "Connecticut",
                "salesperson": "John Doe"
            },

然而，“销售人员”仍然处于“价值观”。 我试过了

result = (df.groupby(['salesperson'])
            .apply(lambda x: x.to_dict('r'))
            .drop('salesperson')
            .to_json(orient='table')
            )

但这似乎不是正确的方法。

我不知道如何告诉它使用“salesperson”作为索引并将其从输出中删除，而不是在创建后实际编辑JSON文件。

Answer 1

下面的代码删除了不需要的密钥。

第1步：

分配变量。

data = {
        "salesperson": "John Doe",
        "values": [
            {
                "region": "North America",
                "state": "Connecticut",
                "salesperson": "John Doe"
            }]
       }

第2步：删除密钥

del data['salesperson']

输出：

Answer 2

我不得不申请drop申请前的关键to_dict()

result = df.groupby(df.salesperson).apply(
    lambda x: x.drop('salesperson', 1).to_dict('records')).to_json(orient='index')

这会从生成的JSON值中删除密钥，同时将其保留为索引。