[Python知识库]python按标签提取csv数据

# -*- coding:utf-8 -*-
"""
作者：sunli
日期：2022年03月02日9:57
"""
# -*- coding:utf-8 -*-

import csv


input_file = 'D:/shujuji/MobiAct_Dataset_v2.0/kalman/BSC_1_2_annotated.csv'
output_file = 'D:/shujuji/MobiAct_Dataset_v2.0/kalman/0.csv'

with open(input_file, 'r') as csv_in_file:
    with open(output_file, 'a', newline='') as csv_out_file: #可在后面追加
    # with open(output_file, 'a', newline='') as csv_out_file:   #重新写入不追加
        filereader = csv.reader(csv_in_file)
        filewriter = csv.writer(csv_out_file)
        header = next(filereader)
        filewriter.writerow(header)
        for row_list in filereader:
            supplier = str(row_list[6]).strip()
            if supplier == '0':
                print(row_list)
                filewriter.writerow(row_list)