LoginSignup
9
16

More than 3 years have passed since last update.

pandas CSVファイル・バッファの読み書き

Posted at

pandasでCSVファイルの読み込み

pandasのread_csv関数のパラメタがfilepath_or_bufferとなっていたので、
CSVファイルではなくbufferを利用する方法をメモしておく。

ファイルを利用する

よく事例が転がっているタイプ。

CSVファイルからDataFrameに読み込む

import pandas as pd
df = pd.read_csv("in.csv")

DataFrameをCSV形式でファイルに書き込む

df.to_csv("out.csv")

バッファを利用する

バッファを活用するパターンは事例がすぐにみつからない。

バッファからDataFrameに読み込む

import io
import pandas as pd

data = 'AAA,BBB\n1,2\n3,4'
df = pd.read_csv(io.StringIO(data))

DataFrameをCSV形式でバッファに書き込む

buffer = io.StringIO()
df.to_csv(buffer, index=False)    # indexを指定しないと行番号が出力される
print(buffer.getvalue())          # バッファの参照
実行結果
AAA,BBB
1,2
3,4

参考

pandas 0.24.2 documentation » API Reference » Input/Output » read_csv

9
16
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
9
16