将带有公式的 Pandas DataFrames 保存到 xlsx 文件

Question

在 Pandas DataFrame 中，我有一些带有值的“单元格”和一些需要包含 excel 公式的单元格。我读过我可以得到公式

link = 'HYPERLINK("#Groups!A' + str(someInt) + '"; "LINKTEXT")'
xlwt.Formula(link)

并将它们存储在数据框中。

当我尝试将我的数据框保存为 xlsx 文件时

writer = pd.ExcelWriter("pandas" + str(fileCounter) + ".xlsx", engine = "xlsxwriter")
df.to_excel(writer, sheet_name = "Paths", index = False)
# insert more sheets here
writer.save()

我收到错误：

TypeError: Unsupported type <class 'xlwt.ExcelFormula.Formula'> in write()

所以我试图将我的公式作为字符串写入我的数据框，但 Excel 想要恢复文件内容，然后用0's填充所有公式单元格。

编辑：我设法让它与常规字符串一起工作，但仍然对 xlwt 公式的解决方案感兴趣。

所以我的问题是：如何将带有公式的数据帧保存到 xlsx 文件？

Answer 1

由于您使用的是 xlsxwriter，默认情况下会将字符串解析为公式（“strings_to_formulas：启用 worksheet.write() 方法将字符串转换为公式。默认值为 True”），因此您可以简单地将公式指定为数据框中的字符串。

引用数据框中其他列的公式列示例：

d = {'col1': [1, 2], 'col2': [3, 4]}
df = pd.DataFrame(data=d)
writer = pd.ExcelWriter("foo.xlsx", engine="xlsxwriter")
df["product"] = None
df["product"] = (
    '=INDIRECT("R[0]C[%s]", 0)+INDIRECT("R[0]C[%s]", 0)'
    % (
        df.columns.get_loc("col1") - df.columns.get_loc("product"),
        df.columns.get_loc("col2") - df.columns.get_loc("product"),
    )
)
df.to_excel(writer, index=False)
writer.save()

产生以下输出：

Answer 2

使用 编写 df 后table.to_excel(writer, sheet_name=...)，我write_formula()在本示例中使用 as （编辑以添加完整循环）。要在数据框中写入所有公式，请读取数据框中的每个公式。

 # replace the right side below with reading the formula from your dataframe
 # e.g., formula_to_write = df.loc(...)`

 rows = table.shape[0]
 for row_num in range(1 + startrow, rows + startrow + 1):
    formula_to_write = '=I{} * (1 - AM{})'.format(row_num+1, row_num+1) 
    worksheet.write_formula(row_num, col, formula_to_write)`

稍后在代码中（我似乎记得其中之一可能是多余的，但我还没有查过）： writer.save() workbook.close()

文档在这里。