如何将python代码输出存储到csv文件

Question

我想将我的 python 代码结果存储到 csv 文件中，但这是我的 python 代码，我没有在我的 csv 文件中显示我的 python 结果

import csv
import json
from collections import Counter

with open('result.csv', 'w') as output:
    with open('simplejson2.json', 'r') as f:
        str1=f.read()

    output_data=csv.writer(output, delimiter=',')
    data_csv= csv.reader(f, delimiter=',')
    data=json.loads(str1)

    c = Counter(k[:] for d in data for k, v in d.items() if k.startswith('sta') and v)
    output_data.writerow(d)
    print("there are total", c['status'], "test case")

    c = Counter(k[:] for d in data for k, v in d.items() if k.startswith('status') and v.startswith('failed'))
    output_data.writerow(d)

    if c['status'] > 0:
        print("There are", c['status'], "failed cases")
    else:
        print("there are", c['status'], "sucessfully pass")

这是我的 simplejson2.json 文件

[
  {
    "status": "passed",
    "name": "Whiskers",
    "species" : "cat",
    "foods": {
      "likes": ["celery", "strawberries"],
      "dislikes": ["carrots"]
    }
  },
  {
    "status": "failed",
    "name": "Woof",
    "species" : "dog",
    "foods": {
      "likes": ["dog food"],
      "dislikes": ["cat food"]
    }
  },
  {
    "status": "failed",
    "name": "Fluffy",
    "species" : "cat",
    "foods": {
      "likes": ["canned food"],
      "dislikes": ["dry food"]
    }
  }
]

我的python代码输出是：-

总共有 3 个测试用例 有 2 个测试用例失败

Answer 1

您正在尝试将d 保存在未分配到任何地方的 csv 中。您仅在此处分配 d，这在此处之外不可用。或者您想将c 保存在csv 中，但您错误地写了d。

根据评论编辑：

with open('simplejson2.json', 'r') as f:
        str1=f.read()
        data_csv= csv.reader(f, delimiter=',')

您只能在with 中读取f。如果您尝试在with 之外读取f，则该文件已关闭，您将给出该错误。

Answer 2

不知道你想要达到什么目的（或者为什么你需要一个计数器而不是仅仅使用len），但这是你代码的一个工作版本：

import csv
import json
from collections import Counter

with open('result.csv', 'w') as output:
    with open('simplejson2.json', 'r') as f:
        str1=f.read()

        output_data=csv.writer(output, delimiter=',')
        data_csv= csv.reader(f, delimiter=',')
        data=json.loads(str1)

        status_fields = [k[:] for d in data for k, v in d.items() if k.startswith('sta') and v]
        c = Counter(status_fields)
        output_data.writerow(status_fields)
        print("there are total", c['status'], "test case")

        status_fields_failed = [k[:] for d in data for k, v in d.items() if k.startswith('status') and v.startswith('failed')]
        c = Counter(status_fields_failed)
        output_data.writerow(status_fields_failed)

        if c['status'] > 0:
            print("There are", c['status'], "failed cases")
        else:
            print("there are", c['status'], "sucessfully pass")

生成的文件内容result.csv

status,status,status
status,status

所做的更改：

• 缩进
• 固定 for 循环表达式（d 未定义，Counter 构造函数需要可迭代）