csv写入

将List写入csv

利用如下方法，将data保存为一个2行3列的csv文件。

import csvdata_list = []data_list.append([first_number, second_number, thrid_number])data_list.append([first_string, second_string, thrid_number])with open("./saved_data/data.csv", "w", newline="") as f:writer = csv.writer(f)writer.writerow(["name1", "name2", "name3"]) # name of columnwriter.writerows(data_list)

注意writerow和writerows的区别

csv读取

普通按行读取

import csv# 读取csvcsv_file = open("all_library.csv", "r")reader = csv.reader(csv_file)for line in reader:# 忽略第一行if reader.line_num == 1:continuegithub_url = line[2] # 获取该行第二列的数值csv_file.close()

若要将csv读取为矩阵：

import pandas as pdrisk_data = pd.read_csv("risk_factors.csv") # 若第一列为序号，则可在参数中添加 `, index_col=0`y = risk_data.Biopsy.valuesX = risk_data.drop("Biopsy", axis=1)features = X.columnsX = X.as_matrix()

参考链接

如果觉得《python写入和读取csv文件》对你有帮助，请点赞、收藏，并留下你的观点哦！