科学の箱

科学・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

タプルの操作

タプルはPythonで提供されているデータ構造の一つ。タプルは固定長で変更できない複数の値の集合である。 タプルオブジェクト (tuple object) — Python 3.8.0 ドキュメント …

no image

tensorflowをpipでインストールするときのオプション

tensorflow環境を構築するときにはpipを利用すると簡単である。pipの簡単な使い方としてはinstallオプションを指定すればよい。 pip install tensoflow しかしこのオ …

no image

配列の属性を確認

numpyでは生成した配列の中身を確認するための属性が用意されている。 主な属性は以下の通り 次元数を確認:ndarray.ndim 各次元の大きさを確認:ndarray.shape 全部の要素数:n …

no image

配列をリストから生成

配列をリストから生成 numpyにおける配列はnp.arrayを使って生成する。最も基本となる方法はnp.arrayに引数としてリストを渡してやる方法である。 まずは整数の配列を生成してみる。 pri …

no image

pythonで配列に関係する型

pythonで配列に関連する型は3つある。 list array numpy.ndarray pandas.dataframe list 違っている複数のデータ型を要素として保持できる 多次元配列可能 …

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

side bar top



アーカイブ

カテゴリー