Python Pandas 名称错误 : StringIO is not defined

Question

我无法读取 Pandas 中的数据:输入:

import pandas as pd

data = 'a,b,c\n1,2,3\n4,5,6'

pd.read_csv(StringIO(data),skipinitialspace=True)

输出:

NameError:name 'StringIO' is not defined

请告诉我错误发生的原因，并让我知道要导入什么。

Answer 1

在这里找到解决方案:

错误发生是因为我没有导入StringIO。与 Python 2 不同，在 Python 3 中您需要导入它。

from io import StringIO

导入后没有出现错误。上述问题的输出是:

   a b c
0  1 2 3
1  4 5 6

它也可以从适用于 Python 2 和 3 的 pandas.compat 导入。

from pandas.compat import StringIO