使用嵌套数组将 Csv 转换为 JSON

Question

我有一个 CSV 文件

group, first, last
fans, John, Smith
fans, Alice, White
students, Ben, Smith
students, Joan, Carpenter
...

输出 JSON 文件需要这种格式：

[
{
  "group" : "fans",
  "user" : [
    {
      "first" : "John",
      "last" :  "Smith"
    },
    {
      "first" : "Alice",
      "last" :  "White"
    }
  ]
},
{
  "group" : "students",
  "user" : [
    {
      "first" : "Ben",
      "last" :  "Smith"
    },
    {
      "first" : "Joan",
      "last" :  "Carpenter"
    }
  ]
}
]

Answer 1

简答
使用itertools.groupby，如documentation 中所述。

长答案
这是一个多步骤的过程。

首先将您的 CSV 转换为 list 或 dict：

from csv import DictReader
with open('data.csv') as csvfile:
    r = DictReader(csvfile, skipinitialspace=True)
    data = [dict(d) for d in r]

groupby需要排序后的数据，所以定义一个函数来获取key，然后像这样传入：

def keyfunc(x):
    return x['group']

data = sorted(data, key=keyfunc)

最后，致电groupby，提供您的排序数据和您的关键功能：

from itertools import groupby
groups = []
for k, g in groupby(data, keyfunc):
    groups.append({
        "group": k,
        "user": [{k:v for k, v in d.items() if k != 'group'} for d in list(g)]
    })

这将遍历您的数据，并且每次密钥更改时，它都会进入for 块并执行该代码，提供k（该组的密钥）和g（dict属于它的对象）。在这里，我们只是将它们存储在一个列表中以备后用。

在此示例中，user 键使用一些非常密集的推导从 user 的每一行中删除 group 键。如果你能忍受那一点点额外的数据，那整行可以简化为：

"user": list(g)

结果如下：

[
  {
    "group": "fans",
    "user": [
      {
        "first": "John",
        "last": "Smith"
      },
      {
        "first": "Alice",
        "last": "White"
      }
    ]
  },
  {
    "group": "students",
    "user": [
      {
        "first": "Ben",
        "last": "Smith"
      },
      {
        "first": "Joan",
        "last": "Carpenter"
      }
    ]
  }
]