Pythonでcsvファイルを読み書きする方法を現役エンジニアが解説【初心者向け】

初心者向けにPythonでcsvファイルを読み書きする方法について現役エンジニアが解説しています。csvファイルは、データがカンマなどの文字で区切られているものです。Pythonでは標準のcsvモジュールを使って読み書きします。open、reader、writerメソッドを使ってみましょう。

TechAcademyマガジンはオンラインのプログラミングスクールTechAcademy [テックアカデミー]が運営する教育×テクノロジーのWebメディアです。初心者でもすぐ勉強できる記事が2,000以上あります。

Pythonでcsvファイルを読み書きする方法について解説します。

Pythonについてそもそもよく分からないという方は、Pythonとは何なのか解説した記事をまずご覧ください。

 

なお本記事は、TechAcademyのPythonオンライン講座の内容をもとにしています。

 

田島悠介

今回は、Pythonに関する内容だね!

大石ゆかり

どういう内容でしょうか?

田島悠介

Pythonでcsvファイルを読み書きする方法について詳しく説明していくね!

大石ゆかり

お願いします!

 

csv.readerの使い方

csvファイルとは、テキストファイルの1種で、ファイルの内部のデータがカンマで区切られて保持されているものです。
 

5.1,3.5,1.4,0.2,Iris-setosa
4.9,3.0,1.4,0.2,Iris-setosa
4.7,3.2,1.3,0.2,Iris-setosa
4.6,3.1,1.5,0.2,Iris-setosa
5.0,3.6,1.4,0.2,Iris-setosa

 
今回の記事ではcsvファイルの例として、機械学習で良く使われるIrisを使用します。

以下のサイトの「iris.data」をクリックしてファイルをダウンロードして、ファイル名を「iris.csv」にリネームしておきましょう。

Index of /ml/machine-learning-databases/iris

Pythonの標準ライブラリでcsvファイルの読み込みと書き込みを行えるのがcsvモジュールです。
csvモジュールはファイルの読み込みまたは書き込みを行うファイルオブジェクトと共に使用します。

ファイルの読み書きについては以下の記事も参考にしてください。

Pythonでファイルに書き込む方法【初心者向け】

csvファイルを読み込むにはcsv.readerを使用します。
csv.readerを使うことで、csvファイルを1行ずつ、list型で取得できます。それぞれの項目の型は文字列となります。
 

import csv

file = open('iris.csv', 'r', encoding = "utf_8")
reader = csv.reader(file)
for row in reader:
? # csvファイル1行ずつの処理を記述
file.close()

 

csvファイルの中身を出力する

それでは実際にcsvファイルの中身を出力してみましょう。
 

file = open('iris.csv', 'r', encoding = "utf_8")
reader = csv.reader(file)
for row in reader:
  print(row)
file.close()

 
実行結果は以下のようになります(一部)。
 

['5.1', '3.5', '1.4', '0.2', 'Iris-setosa']
['4.9', '3.0', '1.4', '0.2', 'Iris-setosa']
['4.7', '3.2', '1.3', '0.2', 'Iris-setosa']
['4.6', '3.1', '1.5', '0.2', 'Iris-setosa']
['5.0', '3.6', '1.4', '0.2', 'Iris-setosa']

 

[PR] Pythonで挫折しない学習方法を動画で公開中

csv.writerの使い方

csvファイルに書き込みを行うにはcsv.writerを使用します。
 

file = open('iris2.csv', 'w', encoding = "utf_8")
writer = csv.writer(file)
# csvファイルに書き込む処理を記述
file.close()

 

csvファイルに文字列を書き込みする

最後に、irisデータをリストに5行分だけ読み込んで、別のcsvファイルに書き込む例をご紹介します。
 

import csv

# 空のリストを作成
iris = []

# 空のリストにcsvファイルを5行読み込み
file = open('iris.csv', 'r', encoding = "utf_8")
reader = csv.reader(file)
for row in reader:
  iris.append(row)
  if(len(iris)>=5):
    break
file.close()

# リストの内容をcsvファイルに書き込み
file = open('iris2.csv', 'w', encoding = "utf_8")
writer = csv.writer(file)
for row in iris:
  writer.writerow(row)
file.close()

 
実行結果は以下のようになります。iris2.csvというファイルが作成されているのが分かります。
 

5.1,3.5,1.4,0.2,Iris-setosa
4.9,3.0,1.4,0.2,Iris-setosa
4.7,3.2,1.3,0.2,Iris-setosa
4.6,3.1,1.5,0.2,Iris-setosa
5.0,3.6,1.4,0.2,Iris-setosa

 

監修してくれたメンター

太田和樹(おおたかずき)

ITベンチャー企業のPM兼エンジニア

普段は主に、Web系アプリケーション開発のプロジェクトマネージャーとプログラミング講師を行っている。守備範囲はフロントエンド、モバイル、サーバサイド、データサイエンティストと幅広い。その幅広い知見を生かして、複数の領域を組み合わせた新しい提案をするのが得意。

開発実績:画像認識技術を活用した駐車場混雑状況把握(実証実験)、音声認識を活用したヘルプデスク支援システム、Pepperを遠隔操作するアプリの開発、大規模基幹系システムの開発・導入マネジメント。

地方在住。仕事のほとんどをリモートオフィスで行う。通勤で消耗する代わりに趣味のDIYや家庭菜園、家族との時間を楽しんでいる。

 

大石ゆかり

内容分かりやすくて良かったです!

田島悠介

ゆかりちゃんも分からないことがあったら質問してね!

大石ゆかり

分かりました。ありがとうございます!

 

オンラインのプログラミングスクールTechAcademyではPythonを使って機械学習の基礎を学ぶPythonオンライン講座を開催しています。

初心者向けの書籍を使って人工知能(AI)や機械学習について学ぶことができます。

現役エンジニアがパーソナルメンターとして受講生に1人ずつつき、マンツーマンのメンタリングで学習をサポートし、最短4週間で習得することが可能です。

また、現役エンジニアから学べる無料のプログラミング体験会も実施しているので、ぜひ参加してみてください。