pandasでCSVファイルを読み込み・書き込みする

pandasでCSVファイルを読み込み・書き込みする方法をまとめました。

IrisデータセットのCSVファイルの読み込み・書き込みを例として解説します。

CSVファイルを読み込む

pandasでCSVファイルを読み込むときは、read_csv()を使用します。

CSVファイルが置かれているパスを引数として渡すことで、CSVファイルを読み込むことができます。

下記はIrisデータセットのCSVファイルを読み込んで、CSVファイルの内容を確認しています。

import pandas as pd

df = pd.read_csv('./iris-dataset.csv')
df.head()

    sepal_length sepal_width petal_length petal_width species
0            5.1         3.5          1.4         0.2  setosa
1            4.9         3.0          1.4         0.2  setosa
2            4.7         3.2          1.3         0.2  setosa
3            4.6         3.1          1.5         0.2  setosa
4            5.0         3.6          1.4         0.2  setosa

CSVファイルを書き込む

pandasでCSVファイルを書き込むときは、to_csv()を使用します。

CSVファイルの出力先のパスを引数として渡すことで、CSVファイルを書き込むことができます。

新規で書き込む場合は、パスに指定したCSVファイル名でCSVファイルが作成されます。

既に同名のCSVファイルが存在する場合は、新たに書き込んだ内容が上書きされます。

下記はIrisデータセット内のラベルを抽出して、新規のCSVファイルに書き込み、書き込んだ内容を確認しています。

labels = df.loc[:,'species']
labels.head()

0    setosa
1    setosa
2    setosa
3    setosa
4    setosa
Name: species, dtype: object

labels.to_csv("./iris-labels.csv")

df = pd.read_csv('./iris-labels.csv')
df.head()

    Unnamed: 0  species
0            0   setosa
1            1   setosa
2            2   setosa
3            3   setosa
4            4   setosa

参考

• PandasでCSVファイルを読み書き | DATUM STUDIO株式会社
• Irisデータセットのダウンロード | Excel VBAでIris分類問題
• pandasで任意の位置の値を取得・変更するat, iat, loc, iloc