python-3.x - pandas pd.read_table 支持 io.BytesIO 和 StringIO 吗？

Question

我有一个 io.BytesIO 对象 iostream，它是从磁盘读取的 be2 文件，我将把列标题附加到表/iostream，

f = io.BytesIO()
f.write(b'A,B,C,D\n')
f.write(iostream.getvalue())

pd.read_table(f, sep=',', index_col=False, error_bad_lines=False, encoding='utf-8', dtype=type_map)

但它给了我一个错误，

pandas.errors.EmptyDataError: No columns to parse from file

我想知道如何解决这个问题。

也尝试过

f = io.StringIO()
f.write('A,B,C,D\n')    
f.write(iostream.getvalue().decode())

pd.read_table(f, sep=',', index_col=False, error_bad_lines=False, encoding='utf-8', dtype=type_map)

出现错误

pandas.errors.ParserError: Error tokenizing data. C error: Calling read(nbytes) on source failed. Try engine='python'.

Answer 1

我成功地重现了您的错误。您第一次尝试时遇到的问题是，在调用“pd.read_table”时，您位于流“f”的末尾，因为您刚刚写入了所有内容。 'pd.read_table' 在内部调用 read()，它从当前位置读取。所以它返回一个空字符串。这是错误的原因:

 pandas.errors.EmptyDataError: No columns to parse from file

解决办法很简单。您只需使用“seek”再次移动到流的开头即可。这段代码对我有用:

f = io.BytesIO()
f.write(b'A,B,C,D\n')
f.write(iostream.getvalue())
f.seek(0)

pd.read_table(f, sep=',', index_col=False, error_bad_lines=False, encoding='utf-8')