Python CSVファイル 読み込みと書き込み

背景

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[[],[],[]]