python读取行并将其写入文件(如果列中包含特定值)

2024-09-29 23:23:58 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在读取一个csv,只想将列中具有特定值的行写入另一个输出csv。你知道吗

这是我的密码:

import csv
import pandas as pd

with open('contribs_01.csv', 'r') as file:
    csv_reader = csv.DictReader(file)

    with open('01_output.csv', 'w') as new_file:
        select_fields = ['contributorPersentTypeCd', 'contributorNameLast', 'contributorNameFirst', 'contributorStreetCity', 'contributorStreetStateCd', 'contributorStreetPostalCode', 'contributorOccupation', 'contributorEmployer', 'filerName', 'contributionAmount']

        csv_writer = csv.DictWriter(new_file, fieldnames=select_fields, extrasaction='ignore')

        csv_writer.writeheader()

        for line in csv_reader:
            if pd.np.where(csv_writer.contributorPersentTypeCd.str.contains('INDIVIDUAL'):
                csv_writer.writerow(line)

这是我得到的错误:

  File "01_donor_read.py", line 17
    if pd.np.where(csv_writer.contributorPersentTypeCd.str.contains('INDIVIDUAL'):
                                                                                 ^
SyntaxError: invalid syntax

Tags: csvimportfieldsnewifaswithline

热门问题