python csv 文件學習 小記

python csv 文件學習 小記

python 將list 寫成csv

import pandas as pd
columns = ['PEID', 'name', 'data']
data = [[1,"zhang","2018"],[4,"wang","2018"],[7,"sun","2017"]]
save_file = pd.DataFrame(columns=columns, data=data)
save_file.to_csv('temp.csv', index=False, encoding="utf-8")

python 讀入csv

import csv 
file_name = "temp.csv"
f = open(file_name, 'r')
csvreader = csv.reader(f)
final_list = list(csvreader)
print(final_list)
final_list.pop(0)
print(final_list)

得到結果：

[['PEID', 'name', 'data'], ['1', 'zhang', '2018'], ['4', 'wang', '2018'], ['7', 'sun', '2017']]
[ ['1', 'zhang', '2018'], ['4', 'wang', '2018'], ['7', 'sun', '2017']]

利用pandas直接得到一列

import pandas as pd
df = pd.read_csv("temp.csv")
name_list = list(df["name"])
print(name_list)

得到結果：

["zhang","wang","sun"]

參考文獻：
python中dataframe常見操作：取行、列、切片、統計特徵值