【Py】Pandas使用指南

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