python - 从字符串创建 Pandas DataFrame

Question

为了测试一些功能，我想从一个字符串创建一个 DataFrame。假设我的测试数据如下所示:

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

将这些数据读入 Pandas DataFrame 的最简单方法是什么？

Answer 1

一个简单的方法是使用 StringIO.StringIO (python2)或 io.StringIO (python3)并将其传递给 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=";")