背景
CSVファイルの読み込みと書き込みの方法はネットにたくさんあるんですが、
毎回自分がググるのがつらいので、備忘録を作成します。
pandasライブラリを使ってもいいんですが、
csvファイルのフォーマット自体が整っていないときに使用できるかなと思います。
フォーマットが整っていたら、pandasを使うと楽です。
###CSVファイルの読み込み
read_csv_file.py
import csv
#csvファイル
csv_file = "demo.csv"
with open(csv_file) as f:
reader = csv.reader(f)
#内包表記 1行ごとのデータをlist_に入れる
list_ = [num for num in reader]
#出力
print("list_ ", list_ )
###CSVファイルの書き込み
read_csv_file.py
import csv
#csvファイル
csv_file = "demo.csv"
#newline=""で書き込み時の改行を阻止
with open(csv_file , 'w', newline="") as f:
writer = csv.writer(f)
# writerow writerows の二種類を使い分ける
writer.writerow(list_1)#list_1[A,B,C,D]
writer.writerows(list_2)#list_2[[],[],[]]