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()
参考链接