科学の箱

科学・IT・登山の話題

Python

StringIOを使ってread_csvのテストをしてみる

投稿日:

pandasで操作できるファイルは様々なあるが、csvはJSONと並んで取り扱うことが多いファイルである。ちょっとしたコードの検証をしたいときに毎回csvを用意して、read_csv()で読み込むのはかったるい。このような時StringIOを使うとすべてコード上で完結する。

 

このページの内容

  • StringIOの基本情報
  • StringIOとread_csv
  • パラメータを変えて読み込む

StringIO

StringIOはpythonで用意されている標準オブジェクトである。

class io.StringIO

引数

  • 第一引数: 文字列

戻り値

  • StringIO: このオブジェクトはread_csv()に対して入力として利用できる。

StringIOとread_csv

StringIOに文字列を与えると、ファイル入出力として使える。

from io import StringIO
import pandas as pd
data = ('name,company,city,tel\n'
       'Suzuki,CompanyA,Tokyo,090-123-456\n'
       'Yamada,CompanyB,Chiba,03-123-456\n'
       'Takahashi,CompanyC,Kanagawa,070-123-456')
df=pd.read_csv(StringIO(data))
print(df)
#        name   company      city          tel
0     Suzuki  CompanyA     Tokyo  090-123-456
1     Yamada  CompanyB     Chiba   03-123-456
2  Takahashi  CompanyC  Kanagawa  070-123-456

 

パラメータを変えて読み込む

いくつかパラメータを変えて読み込んでみる

まずはindex_colを指定して、indexを設定する。

df=pd.read_csv(StringIO(data), index_col="name")
print(df)

#             company      city          tel
name                                      
Suzuki     CompanyA     Tokyo  090-123-456
Yamada     CompanyB     Chiba   03-123-456
Takahashi  CompanyC  Kanagawa  070-123-456

 

usecolsを指定して、読み込む列を選ぶ。

df=pd.read_csv(StringIO(data), usecols=["name","city"])
print(df)

#         name      city
0     Suzuki     Tokyo
1     Yamada     Chiba
2  Takahashi  Kanagawa

 

メタ情報

inarticle



メタ情報

inarticle



-Python
-

執筆者:


comment

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です

関連記事

no image

flip, fliplr, flipudを使って配列要素を上下左右、左右、上下反転する

画像処理などでは要素の値を配列全体で反転したいときがある。このようなときに役に立つのがflip, fliplr, flipudである。 flipのマニュアル fliprのマニュアル flipudのマニ …

no image

dataframeで条件を付けて要素を返す方法

前回の続きから。 dfは現在以下のようになっている。 W X Y Z A 2.706850 0.628133 0.907969 0.503826 B 0.651118 -0.319318 -0.848 …

no image

初めてのmatplotlib

matplotlibを使えばpythonで様々なグラフをデータから作成できる。初めてのmatplotlibということで、単純なグラフを作成する。 まずはmatplotlibを読み込む >> …

no image

配列同士の四則演算

ndarray同士で四則演算ができる。 この四則演算自体はnumpyにおけるbroadcastingと呼ばれる機能およびuniversal functionを利用して実現している。 numpyのbro …

no image

Pillowのインストールは成功しているのにエラーが出る

from . import _imaging as core ImportError: DLL load failed: The specified module could not be found …

2019年9月
« 8月   10月 »
 1
2345678
9101112131415
16171819202122
23242526272829
30  

side bar top



アーカイブ

カテゴリー