Backto Python Index
Using Pandas
import pandas as pd
#任意的多组列表
a = [1,2,3]
b = [4,5,6]
#字典中的key值即为csv中列名
dataframe = pd.DataFrame({'a_name':a,'b_name':b})
#将DataFrame存储为csv,index表示是否显示行名,default=True
dataframe.to_csv("test.csv",index=False,sep=',')
注意: 这里再 DataFrame 中使用的是字典,所以顺序是随机的,落到文档里可能顺序是 (b_name, a_name)。
Using csv module
import csv
# write
rows = [['1', '2', '3'], ['4', '5', '6']]
with open('my.csv', 'w+', newline='') as csv_file:
writer = csv.writer(csv_file)
for row in rows:
writer.writerow(row)
# read
with open('test.csv','r') as csv_file:
reader =csv.reader(csv_file) # read line by line
next(reader) # Skip header
for row in reader:
print(row) # already split into list[str] by ','
更灵活的用法,整块读取,或者几行读一次,这样的可以用 pandas 中文件的读写——read_csv() 读取文件,但是读进来的数据按 pd 格式是列存储
的。
Ref
- 10 min to pandas
- python写入csv文件的几种方法总结: 最简便的是 Pandas