Python Tips: 讀取 & 輸出 CSV 教學

CSV 是常用到的結構化資料交換格式, 學會讀取跟寫入 CSV,  對於常常需要和 data 為伍的工程師, 非常受用.

(沿用上一篇 Python Tips: 獲取 API 資料 的數據資料)

• 寫入: 寫入資料前, 需要先定義header, 方便之後 csv 再輸入到其他地方時, 有表頭可以對其.

 # 寫入 csv
 import csv
 
 def write_csv(header, file_name, data):
    with open(file_name, 'w', newline='') as f:
        write = csv.writer(f)
        write.writerow(header)
        write.writerows(data)
 
 file_name = 'api_data.csv'
 header = ['From', 'To', 'Rate', 'timestamp']
 
 output_to_csv = [write_csv(header, file_name, api_data)]

輸入的結果

• 讀取

 # 讀取 csv
 def read_csv(file_name):
    with open(file_name, 'r') as f:
       read_data = csv.reader(f, delimiter=',')
       return [i for i in read_data]
 
 file_name = './api_data.csv'
 from_csv = read_csv(file_name)

讀取結果, 可以發現表頭也被讀取進來了.