【python】CSVファイルの読み書き

csvモジュールを利用したcsvファイルへの書き込みと読み込みのやり方を完全解説！

モジュールコンテンツ

CSVモジュールでは、CSV 形式で書かれたテーブル状のデータを読み書きするためのクラスを実装しています。
この記事ではCSVモジュールを詳しく説明していきます。

関数

import csv

with open('sample.csv',newline='') as f:
    d = csv.reader(f,delimiter=' ',quotechar='|')
    for row in d:
        print(row)

実行結果
['1,a']
['2,b']
['3,c']
['4,d']
['5,e']
['6,f']

import csv

rows = [
    [8,"h"],
    [9,"i"],
    [10,"j"]
]

with open('sample.csv','w',newline='') as f:
    write = csv.writer(f,delimiter=' ',quotechar='|', quoting=csv.QUOTE_MINIMAL)
    write.writerow([7,"g"])
    write.writerows(rows)

7 g
8 h
9 i
10 j

これを利用することでいろいろな表現形式を指定できる。

csv.get_dialect(name)

nameに関連づけられた表現形式を返します。

import csv
csv.register_dialect('pipes', delimiter='|')
l = csv.get_dialect('pipes')
print(l)

<_csv.Dialect object at 0x000001F70FBCDEA0>

csv.register_dialect(name[, dialect[, **fmtparams]])

dialect を name と関連付けます。

7|g
8|h
9|i
10|j

このように区切りが”｜”であるとき、新しい dialect に delimiter を指定して登録できます。

import csv

csv.register_dialect('pipes', delimiter='|') #"|"をpipesというnameで登録

with open('sample.csv', 'r') as f:
    reader = csv.reader(f, dialect='pipes')
    for row in reader:
        print(row)

“｜”を区切り文字として認識してカンマ区切りのファイルのように祖見込まれます。

['7', 'g']
['8', 'h']
['9', 'i']
['10', 'j']

csv.unregister_dialect(name)

nameに関連づけられた表現形式を表現形式レジストリから削除します。

import csv
csv.register_dialect('pipes', delimiter='|')
csv.unregister_dialect('pipes')
l = csv.get_dialect('pipes')
print(l)

_csv.Error: unknown dialect

登録したregisterを読み込もうとしたところエラーが起きたため削除できていることがわかる。

csv.list_dialects()

登録されている全ての表現形式を返します。

import csv

l = csv.list_dialects()
print(l)

['excel', 'excel-tab', 'unix']

csv.filed_size_limit([new_limit])

パーサが許容する現在の最大フィールドサイズを返します。

import csv

f = csv.field_size_limit()
print(f)

131072

クラス

import csv
a = ["年","月"]
with open('sample.csv',newline='') as f:
    d = csv.DictReader(f,fieldnames=a,restkey="末日",dialect='excel')
    for row in d:
        print(row)
    print(d)

{'年': '2020', '月': '1', '末日': ['31']}
{'年': '2020', '月': '2', '末日': ['29']}
{'年': '2020', '月': '3', '末日': ['31']}
{'年': '2020', '月': '4', '末日': ['30']}
{'年': '2020', '月': '5', '末日': ['41']}

import csv

with open('sample.csv','w',newline='') as f:
    fieldnames = ['月','年','末日']
    d = csv.DictWriter(f,fieldnames=fieldnames,restval='NoSet')
    d.writerow({'年':2020,'月':1})
    d.writerow({'年':2020,'月':2})
    d.writerow({'年':2020,'月':3})

1,2020,NoSet
2,2020,NoSet
3,2020,NoSet

'7'|'g'
'8'|'h'
'9'|'i'
'10'|'j'

import csv
sniffer = csv.Sniffer()
sample_bytes = 32
with open('sample.csv','r',newline='') as f:
    dialect = sniffer.sniff(f.read(sample_bytes))
    head = sniffer.has_header(f.read(sample_bytes))
    print(dialect.delimiter)
    print(dialect.quotechar)

    print(head)

|
'
True

定数

csv.QUOTE_ALL

writer オブジェクトに対し、全てのフィールドをクオートするように指示します。

csv.QUOTE_MINIMAL

writer オブジェクトに対し、 delimiter 、 quotechar または lineterminator に含まれる任意の文字のような特別な文字を含むフィールドだけをクオートするように指示します。

csv.QUOTE_NONNUMERIC

writer オブジェクトに対し、全ての非数値フィールドをクオートするように指示します。
reader に対しては、クオートされていない全てのフィールドを float 型に変換するよう指示します。

csv.QUOTE_NONE

writer オブジェクトに対し、フィールドを決してクオートしないように指示します。

現在の delimiter が出力データ中に現れた場合、現在設定されている escapechar 文字が前に付けられます。
escapechar がセットされていない場合、エスケープが必要な文字に遭遇した writer は Error を送出します。
reader に対しては、クオート文字の特別扱いをしないように指示します。

例外

csv.Error

全ての関数において、エラーが検出された際に送出される例外です。

readerオブジェクト

import csv

with open('sample.csv','r',newline='') as f:
    r = csv.reader(f)
    n = r.__next__()
    d = r.dialect
    l = r.line_num
    print('next:',n)
    print('dialect:',d)
    print('line_num:',l)
    
    dr = csv.DictReader(f,fieldnames=['年','月','末日'])
    f = dr.fieldnames
    print('filedname:',f)

next: ['2020', '1', '31']
dialect: <_csv.Dialect object at 0x00000290B64597E0>
line_num: 1
filedname: ['年', '月', '末日']

writerオブジェクト

csvwriter.writerow(row)

現在の方言に従ってフォーマットされた行パラメーターをライターのファイルオブジェクトに書き込みます。

import csv

with open ('sample.csv','w',newline='') as f:
    w = csv.writer(f)
    w.writerow(['a','b','c'])

a,b,c

csvwriter.writerows(rows)

rows引数の全ての要素を現在の表現形式に基づいて書式化し、writer のファイルオブジェクトに書き込みます。

import csv

with open ('sample.csv','w',newline='') as f:
    w = csv.writer(f)
    w.writerows([['a'],['b'],['c']])

a
b
c

csvwriter.dialect

writer で使われる表現形式の読み出し専用の記述です。

import csv

with open ('sample.csv','w',newline='') as f:
    w = csv.writer(f,dialect='excel')
    print(w.dialect)

<_csv.Dialect object at 0x0000024D4AF64FC0>

DictWriter.writeheader()　(DictWriterオブジェクト)

現在の方言に従ってフォーマットされた、（コンストラクターで指定された）フィールド名を含む行をライターのファイルオブジェクトに書き込みます。 内部で使用されるcsvwriter.writerow（）呼び出しの戻り値を返します。

import csv

with open ('sample.csv','w',newline='') as f:
    w = csv.writer(f,dialect='excel')
    print(w.dialect)

a,b,c