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

问题描述

我正在尝试获取一个 csv 文件并遍历它，只获取具有蓝色类别的行并将其写入 JSON 文件。到目前为止，我已经创建了一个程序，它接收所有 csv 行并写入 JSON 文件。我不太确定如何仅选择具有特定值的某些 csv 行。 这是带有此输出的程序

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

输出：

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

以这个csv文件为例

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

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

if row.category == 'blue'

但这显然行不通

解决方法

由于每一行都被 DictReader 转换为 python dict，你只需要在一个位置扩展你的代码。

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

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