Python 標準モジュール「csv」で書き込む代表例2つ

　本記事では、標準モジュールの「csv」でcsvファイルを作成する雛形コードを2つ載せました。いずれも下図のようなカンマ区切りのテキストファイルを出力します。行数が少ない場合はpandasよりも可読性が良いかもしれません。

▼一行ずつ書き込む場合

import csv

header_list = ['A', 'B', 'C']
data_list1 = [1, 2, 3]
data_list2 = [4, 5, 6]

## 一行ずつ書き込む場合
with open('test1.csv', 'w', newline = '') as f:
    writer = csv.writer(f) # オブジェクト生成
    writer.writerow(header_list) # 1行目
    writer.writerow(data_list1) # 2行目
    writer.writerow(data_list2) # 3行目

▼複数行を一度に書き込む場合

import csv

header_list = ['A', 'B', 'C']
data_list1 = [1, 2, 3]
data_list2 = [4, 5, 6]

## 複数行を一度に書き込む場合
# 多重リストを作成する
data_rows_list = []
data_rows_list.append(header_list)
data_rows_list.append(data_list1)
data_rows_list.append(data_list2)
print(data_rows_list)

# 複数行を一度に書き込む
with open('test2.csv', 'w', newline = '') as f:
    writer = csv.writer(f)
    writer.writerows(data_rows_list)

以上

<広告>

リンク

リンク