从字符串创建 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 ？

原文由 Emil H 发布，翻译遵循 CC BY-SA 4.0 许可协议

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=";")

原文由 Emil H 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

分割法

data = input_string
df = pd.DataFrame([x.split(';') for x in data.split('\n')])
print(df)

原文由 Shaurya Uppal 发布，翻译遵循 CC BY-SA 4.0 许可协议