熊猫:将数据框写入excel文件*对象*(不是文件)?

时间:2019-07-10 17:26:35

标签: python pandas

我有一个要转换为excel文件的数据框,并使用HTTP返回它。数据框的to_excel方法接受路径或ExcelWriter,后者又指向路径。

有什么方法可以将数据帧转换为文件对象,而无需将其写入磁盘?

1 个答案:

答案 0 :(得分:0)

这可以使用标准库中的BytesIO对象来完成:

import pandas
from io import BytesIO

# Create Random Data for example
cols = ["col1", "col2"]
df = pandas.DataFrame.from_records([{k: 0.0 for k in cols} for _ in range(25)])

# Create an in memory binary file object, and write the dataframe to it.
in_memory_fp = BytesIO()
df.to_excel(in_memory_fp)

# Write the file out to disk to demonstrate that it worked.
in_memory_fp.seek(0,0)
with open("my_file.xlsx", 'wb') as f:
    f.write(in_memory_fp.read())

在上面的示例中,我将对象写到文件中,以便您可以验证它是否有效。如果您只想将原始二进制数据返回到内存中,则只需:

in_memory_fp.seek(0,0)
binary_xl = in_memory_fp.read()