使用附加关键字/组织将 csv 导入 json/dict答案

【问题标题】：import csv to json/dict with additional keywords/organization使用附加关键字/组织将 csv 导入 json/dict
【发布时间】：2017-05-01 03:43:30
【问题描述】：

我正在尝试将 .csv 转换为 json/dict，以便当前形式的数据：

cat1,cat2,cat3,name
1,2,3,a
4,5,6,b
7,8,9,c

我目前正在使用这样的东西（以及使用 pandas.df bc 导入，它将用于从 json 文件中绘制图形）：

with open('Data.csv') as f:
    reader = csv.DictReader(f)
    rows = list(reader)

print (rows)

[{'cat1': '1', 'name': 'a', 'cat3': '3', 'cat2': '2'}, 
 {'cat1': '4', 'name': 'b', 'cat3': '6', 'cat2': '5'}, 
 {'cat1': '7', 'name': 'c', 'cat3': '9', 'cat2': '8'}]

我希望它在 json/dict 格式中看起来像这样：

{"data: [{"all_cats": {"cat1": 1}, {"cat2": 2}, {"cat3": 3}}, "name": a},
         {"all_cats": {"cat1": 4}, {"cat2": 5}, {"cat3": 6}}, "name": b},
         {"all_cats": {"cat1": 7}, {"cat2": 8}, {"cat3": 8}}, "name": c}]}

直接导入不允许我在 'all_cats' 下包含：'cat1'、'cat2'、'cat3' 并保持 'name' 分开。

任何帮助将不胜感激。

【问题讨论】：

您想要的格式不正确。

标签： json python-3.x csv dictionary

【解决方案1】：

由于它是空格分隔而不是逗号分隔，因此您必须添加delimiter=" "。此外，由于您的某些行事先有空格，这意味着您还必须添加 skipinitialspace=True。

reader = csv.DictReader(f, delimiter=" ", skipinitialspace=True)
rows = list(dict(row) for row in reader)

如果你现在这样做：

for row in rows:
    print(row)

输出将是：

{'cat1': '1', 'cat2': '2', 'cat3': '3', 'name': 'a'}
{'cat1': '4', 'cat2': '5', 'cat3': '6', 'name': 'b'}
{'cat1': '7', 'cat2': '8', 'cat3': '9', 'name': 'c'}

【讨论】：

感谢您的回答。我在这里以这种方式表示数据，因此更容易查看。实际数据是 .csv
@virgiliocyte 您应该将实际数据添加到问题中，而不是这样做只会使回答问题变得复杂。但是，您想要的格式仍然不正确。
@virgiliocyte 现在如果您已经加载了数据，我看不到DictReader 中的意义。
现在应该更正。我是 python 和编码的新手。道歉。

【解决方案2】：

正如在另一个答案中已经提到的，您没有为您想要实现的目标指定有效的 JSON 格式。您可以使用json.loads(jsonDATAstring) 函数检查字符串是否包含有效的 JSON 格式：

import json
jsonDATAstring_1 = """
{"data: [{"all_cats": {"cat1": 1}, {"cat2": 2}, {"cat3": 3}}, "name": a},
         {"all_cats": {"cat1": 4}, {"cat2": 5}, {"cat3": 6}}, "name": b},
         {"all_cats": {"cat1": 7}, {"cat2": 8}, {"cat3": 8}}, "name": c}]}
"""
json.loads(jsonDATAstring_1)

如果你指定了预期的 JSON 格式会导致：

json.decoder.JSONDecodeError: Expecting ':' delimiter: line 2 column 12 (char 12)

根据我从您的问题中了解到的情况，我假设您想要获取的 JSON 字符串是以下字符串：

jsonDATAstring_2 = """
{"data": [{"all_cats": {"cat1": 1, "cat2": 2, "cat3": 3}, "name": "a"},
          {"all_cats": {"cat1": 4, "cat2": 5, "cat3": 6}, "name": "b"},
          {"all_cats": {"cat1": 7, "cat2": 8, "cat3": 8}, "name": "c"}]}
"""
json.loads(jsonDATAstring_2)

第二个字符串加载正常，所以假设：

rows = [{'cat1': '1', 'name': 'a', 'cat3': '3', 'cat2': '2'}, 
        {'cat1': '4', 'name': 'b', 'cat3': '6', 'cat2': '5'}, 
        {'cat1': '7', 'name': 'c', 'cat3': '9', 'cat2': '8'}]

你可以得到你想要的如下：

dctData = {"data": []}
lstCats = ['cat1', 'cat2', 'cat3']
for row in rows:
    dctAllCats  = {"all_cats":{}, "name":"?"}
    for cat in lstCats:
        dctAllCats["all_cats"][cat] = row[cat]
    dctAllCats["name"] = row["name"]
    dctData["data"].append(dctAllCats)

import pprint
pp = pprint.PrettyPrinter()
pp.pprint(dctData)

什么给了：

{'data': [{'all_cats': {'cat1': '1', 'cat2': '2', 'cat3': '3'}, 'name': 'a'},
          {'all_cats': {'cat1': '4', 'cat2': '5', 'cat3': '6'}, 'name': 'b'},
          {'all_cats': {'cat1': '7', 'cat2': '8', 'cat3': '9'}, 'name': 'c'}]}

现在可以将 Python 字典对象序列化为 JSON 字符串（或文件）：

jsonString = json.dumps(dctData)
print(jsonString)

什么给了：

{"data": [{"all_cats": {"cat1": "1", "cat2": "2", "cat3": "3"}, "name": "a"}, {"all_cats": {"cat1": "4", "cat2": "5", "cat3": "6"}, "name": "b"}, {"all_cats": {"cat1": "7", "cat2": "8", "cat3": "9"}, "name": "c"}]}

【讨论】：