Question

我可以将StringIO对象传递给pd.to_csv（）就好了：

io = StringIO.StringIO()
pd.DataFrame().to_csv(io)

但是当使用excel编写器时，我遇到了很多麻烦。

io = StringIO.StringIO()
writer = pd.ExcelWriter(io)
pd.DataFrame().to_excel(writer,"sheet name")
writer.save()

返回

AttributeError: StringIO instance has no attribute 'rfind'

我试图在不调用ExcelWriter的情况下创建pd.ExcelWriter()对象，但遇到了一些麻烦。这是我到目前为止所尝试的：

from xlsxwriter.workbook import Workbook
writer = Workbook(io)
pd.DataFrame().to_excel(writer,"sheet name")
writer.save()

但现在我得到AttributeError: 'Workbook' object has no attribute 'write_cells'

如何将excel格式的pandas数据帧保存到StringIO对象？

Answer 1

Pandas期望ExcelWriter构造函数的文件名路径，尽管每个编写器引擎都支持StringIO。也许这应该在Pandas中作为bug / feature请求提出。

与此同时，这是使用Pandas xlsxwriter引擎的变通方法示例：

import pandas as pd
import StringIO

io = StringIO.StringIO()

# Use a temp filename to keep pandas happy.
writer = pd.ExcelWriter('temp.xlsx', engine='xlsxwriter')

# Set the filename/file handle in the xlsxwriter.workbook object.
writer.book.filename = io

# Write the data frame to the StringIO object.
pd.DataFrame().to_excel(writer, sheet_name='Sheet1')
writer.save()
xlsx_data = io.getvalue()

更新：从Pandas 0.17开始，现在可以更直接地执行此操作：

# Note, Python 2 example. For Python 3 use: output = io.BytesIO().
output = StringIO.StringIO()

# Use the StringIO object as the filehandle.
writer = pd.ExcelWriter(output, engine='xlsxwriter')

另请参阅XlsxWriter文档中的Saving the Dataframe output to a string。

Answer 2

如果你不介意使用xlwt作为你的编写器，那么看一下pandas.io.excel源看起来不应该是一个太大的问题。其他引擎可能也不是那么困难，但是xlwt很容易跳出来，因为它的save方法需要一个流或一个文件路径。

你需要最初传入一个文件名只是为了让pandas高兴，因为它会检查引擎的文件扩展名，以确保它是支持的格式。但是在xlwt引擎的情况下，它只是将文件名填充到对象的path属性中，然后在save方法中使用它。如果将路径属性更改为流，则在调用save方法时，它将很乐意保存到该流。

以下是一个例子：

import pandas as pd
import StringIO
import base64

df = pd.DataFrame.from_csv('http://moz.com/top500/domains/csv')
xlwt_writer = pd.io.excel.get_writer('xlwt')
my_writer = xlwt_writer('whatever.xls')  #make pandas happy 
xl_out = StringIO.StringIO()
my_writer.path = xl_out  
df.to_excel(my_writer)
my_writer.save()
print base64.b64encode(xl_out.getvalue())

这是快速，简单和略微肮脏的方式。顺便说一句......一个更简洁的方法就是继承ExcelWriter（或其中一个现有的子类，例如_XlwtWriter） - 但老实说，更新路径属性所涉及的很少，我投票给你展示简单的方法，而不是去稍长的路线。

Answer 3

对于那些不使用xlsxwriter作为engine=的{{1}}的人来说，这是一个在内存中使用to_excel的解决方案：

openpyxl

说明：in_memory_file = StringIO.StringIO() xlw = pd.ExcelWriter('temp.xlsx', engine='openpyxl') # ... do many .to_excel() thingies xlw.book.save(in_memory_file) # if you want to read it or stream to a client, don't forget this in_memory_file.seek(0)包装类通过ExcelWriter属性公开引擎单个工作簿。对于.book，您可以照常使用openpyxl方法！

Answer 4

这些都不是我的工作。我有一个观点，我想从Django返回一个Excel工作簿。我找到了解决方法from the pandas documentation。

import io
bio = io.BytesIO()
writer = pd.ExcelWriter(bio, engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1')
writer.save()
bio.seek(0)

# BONUS CONTENT
# .. because I wanted to return from an api
response = HttpResponse(bio, content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
response['Content-Disposition'] = 'attachment; filename=myfile.xlsx'
return response # returned from a view here

注意，我使用该值作为内容类型，因为根据mozzilla文档，它是mime类型。来自以下链接中的“ .xlsx”。根据需要更换。 https://developer.mozilla.org/en-US/docs/Web/HTTP/Basics_of_HTTP/MIME_types/Common_types

使用Pandas Excelwriter写入StringIO对象？

4 个答案: