Python 문자열에서 판다스 데이터프레임 생성하기, Create Pandas DataFrame from a string

질문

기능을 테스트하기 위해 문자열에서 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 (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=";")