使用 python 从 BytesIO 创建一个 excel 文件

Create an excel file from BytesIO using python

我正在使用 pandas 库将 excel 存储到 bytesIO 内存中。稍后,我将这个 bytesIO 对象存储到 SQL 服务器中,如下所示-

    df = pandas.DataFrame(data1, columns=['col1', 'col2', 'col3'])
    output = BytesIO()
    writer = pandas.ExcelWriter(output,engine='xlsxwriter')
    df.to_excel(writer)
    writer.save()
    output.seek(0)
    workbook = output.read()

    #store into table
    Query = '''
            INSERT INTO [TABLE]([file]) VALUES(?)
            '''
    values = (workbook)
    cursor = conn.cursor()
    cursor.execute(Query, values)
    cursor.close()
    conn.commit()

   #Create excel file.
   Query1 = "select [file] from [TABLE] where [id] = 1"
   result = conn.cursor().execute(Query1).fetchall()
   print(result[0])

现在,我想从 table 中拉回 BytesIO 对象并创建一个 excel 文件并将其存储在本地。我该怎么做?

最后,我得到了 solution.Below 执行的步骤:

  1. 获取Dataframe并将其转换为excel并以BytesIO格式存储在内存中。
  2. 在具有 varbinary(max) 的数据库列中存储 BytesIO 对象
  3. 拉取存储的 BytesIO 对象并在本地创建一个 excel 文件。

Python代码:

#Get Required data in DataFrame:
df = pandas.DataFrame(data1, columns=['col1', 'col2', 'col3'])

#Convert the data frame to Excel and store it in BytesIO object `workbook`:
output = BytesIO()
writer = pandas.ExcelWriter(output,engine='xlsxwriter')
df.to_excel(writer)
writer.save()
output.seek(0)
workbook = output.read()

#store into Database table
Query = '''
        INSERT INTO [TABLE]([file]) VALUES(?)
        '''
values = (workbook)
cursor = conn.cursor()
cursor.execute(Query, values)
cursor.close()
conn.commit()

#Retrieve the BytesIO object from Database
Query1 = "select [file] from [TABLE] where [id] = 1"
result = conn.cursor().execute(Query1).fetchall()

WriteObj = BytesIO()
WriteObj.write(result[0][0])  
WriteObj.seek(0)  
df = pandas.read_excel(WriteObj)
df.to_excel("outputFile.xlsx")