Question

我有一个excel文件，其中数据以csv格式保存。这个数据存在于excel文件中，如下所示，在A列下面（CSV文件由我编写的LabView软件代码生成生成数据）。我还附上了csv文件的图像，以供我在问题的结尾处参考。

RPM,Load Current,Battery Output,Power Capacity
1200,30,12,37
1600,88,18,55

我想以这种格式创建一个Json文件

{  
   "power_capacity_data" :
   {
   "rpm" : ["1200","1600"],
   "load_curr" : ["30","88"],
   "batt_output" : ["12","18"],
   "power_cap" : ["37","55"]
   }
 }

这是我的代码

import csv
import json 

def main():

    #created a dictionary so that i can append data to it afterwards

    power_data = {"rpm":[],"load_curr":[],"batt_output":[],"power_cap":[]}

    with open('power1.lvm') as f:
        reader = csv.reader(f)

        #trying to append the data of column "RPM" to dictionary
       rowcount = 0
       for row in reader:   
           if rowcount == 0:
               #trying to skip the first row
               rowcount = rowcount + 1 
           else:
               power_data['rpm'].append(row[0])
               print(row)

        json_report = {}        
        json_report['pwr_capacity_data'] = power_data
        with open('LVMJSON', "w") as f1:
            f1.write(json.dumps(json_report, sort_keys=False, indent=4, separators=(',', ': '),encoding="utf-8",ensure_ascii=False))
            f1.close()


 if __name__ == "__main__":
     main()

我得到的输出json文件是这样的:(请忽略我的代码中的print（row）语句）

{
"pwr_capacity_data": 
  {
    "load_curr": [],
    "rpm": [
        "1200,30,12.62,37.88",
        "1600,88,18.62,55.88"
    ],
    "batt_output": [],
    "power_cap": []
  } 
}

整行都会保存在列表中，但我只想保存RPM列下的值。有人可以帮我解决我可能做错的事情。谢谢。I have attached an image of csv file to just in case it helps < / p>

Answer 1

你可以使用Python的defaultdict来使它更容易一些。也是一个用于映射所有标题值的字典。

from collections import defaultdict    
import csv
import json

power_data = defaultdict(list)

header_mappings = {
    'RPM' : 'rpm',
    'Load Current' : 'load_curr',
    'Battery Output' : 'batt_output',
    'Power Capacity' : 'power_cap'}

with open('power1.lvm', newline='') as f_input:
    csv_input = csv.DictReader(f_input)

    for row in csv_input:
        for key, value in row.items():
            power_data[header_mappings[key]].append(value)

with open('LVMJSON.json', 'w') as f_output:            
    json.dump({'power_capacity_data' : power_data}, f_output, indent=2)

为您提供输出JSON文件，如下所示：

{
  "power_capacity_data": {
    "batt_output": [
      "12",
      "18"
    ],
    "power_cap": [
      "37",
      "55"
    ],
    "load_curr": [
      "30",
      "88"
    ],
    "rpm": [
      "1200",
      "1600"
    ]
  }
}

将csv转换为json时，未在python中获得预期的输出

1 个答案: