如何遍历 CSV 文件并仅将具有特定值的 csv 行写入 JSON 中的另一个文件？

Question

我正在尝试获取 csv 文件并对其进行迭代，并且仅获取具有蓝色类别的行并将其写入 JSON 文件。 到目前为止，我已经创建了一个程序，它将所有 csv 行写入 JSON 文件。 我不太清楚如何 select 只有某些 csv 线具有一定的价值。 这是带有此 output 的程序

import csv
import json


def csv_to_json(csvFilePath, jsonFilePath):
    jsonArray = []

    with open(csvFilePath, encoding='utf-8') as csvf:
        csvReader = csv.DictReader(csvf)

        for row in csvReader:
            jsonArray.append(row)

    with open(jsonFilePath, 'w', encoding='utf-8') as jsonf:
        jsonString = json.dumps(jsonArray, indent=4)
        jsonf.write(jsonString)


csvFilePath = r'csv_example.csv'
jsonFilePath = r'reader.json'
csv_to_json(csvFilePath, jsonFileP

output：

[
    {
        "id": "1",
        "category": "blue",
        "password": "hi"
    },
    {
        "id": "2",
        "category": "red",
        "password": "no"
    },
    {
        "id": "3",
        "category": "blue",
        "password": "lol"
    },
    {
        "id": "4",
        "category": "yellow",
        "password": "yo"
    },
    {
        "id": "5",
        "category": "yellow",
        "password": "sup"
    },
    {
        "id": "6",
        "category": "orange",
        "password": "hello"
    },
    {
        "id": "7",
        "category": "blue",
        "password": "yeah"
    }
]

以这个 csv 文件为例

"id","category","password"
"1","blue","hi"
"2","red","no"
"3","blue","lol"
"4","yellow","yo"
"5","yellow","sup"
"6","orange","hello"
"7","blue","yeah"

我将如何使它只有具有蓝色类别的行被写入 JSON 文件。 我试过了

if row.category == 'blue'

但这显然行不通

Answer 1

由于每一行都由 DictReader 转换为 python 字典，因此您只需在一个 position 处扩展您的代码。

for row in csvReader:
   if row["category"] != "blue":
      continue
   jsonArray.append(row)

if row["category"]:= "blue":检查该行中关键类别的值是否不是蓝色的。 如果该行不是蓝色的，代码将continue下一行