1. ホーム
  2. python

[解決済み] 文字列からPandas DataFrameを作成する

2022-03-26 05:06:59

質問

ある機能をテストするために DataFrame を文字列から生成します。私のテストデータが次のようなものだとしましょう。

TESTDATA="""col1;col2;col3
1;4.4;99
2;4.5;200
3;4.7;65
4;3.2;140
"""

そのデータをPandasの DataFrame ?

解決方法は?

簡単な方法としては StringIO.StringIO (パイソン2) または io.StringIO (パイソン3) に渡し、それを pandas.read_csv 関数を使用します。例

import sys
if sys.version_info[0] < 3: 
    from StringIO import StringIO
else:
    from io import StringIO

import pandas as pd

TESTDATA = StringIO("""col1;col2;col3
    1;4.4;99
    2;4.5;200
    3;4.7;65
    4;3.2;140
    """)

df = pd.read_csv(TESTDATA, sep=";")