python处理csv文件 常用操作

读取csv并操作

import pandas as pd

csv_path = "report.csv"

# 选择要读取的列的列名 
# usecols可以选择要读取的列,不选的话就全读
df = pd.read_csv(csv_path, usecols=["patient_id", "report"])
# 要存放csv的dict
patient = dict()

# 通过index, row, df.iterrows() 
for index, row in df.iterrows():
	# row["列名"]可以选某一行的某一列内容
    id = row["patient_id"]
    report = str(row["report"])
    if id not in patient.keys():
        patient[id] = ""
    patient[id] += report

写入数据结构到csv操作:

import csv
# data也可以是dict
data = []
with open('test.csv', 'w', newline='') as csvfile:
    writer  = csv.writer(csvfile)
    for row in data:
        writer.writerow(row)

你可能感兴趣的:(python,开发语言,机器学习)