pandas - ModuleNotFoundError : No module named 'xlsxwriter' in databricks

Question

我正在尝试将 pandas 数据框的内容保存到 windows/azure 数据 block 中的 excel 文件中。将 Pandas 导入为 pd

根据数据创建 Pandas 数据框。

df = pd.DataFrame({'Data': [10, 20, 30, 20, 15, 30, 45]})

使用 XlsxWriter 作为引擎创建 Pandas Excel 编写器。

writer = pd.ExcelWriter('pandas_simple.xlsx', engine='xlsxwriter')

将数据框转换为 XlsxWriter Excel 对象。

df.to_excel(writer, sheet_name='Sheet1')

关闭 Pandas Excel 编写器并输出 Excel 文件。

writer.save()

错误>>

ModuleNotFoundError: 没有名为“xlsxwriter”的模块

在第 2 行 pd.ExcelWriter()

databricks 集群在 spark 2.4.4 上运行有关如何解决此问题的任何建议？

Answer 1

确保您已安装 XlsxWriter

 pip install XlsxWriter

您可能需要重新启动内核

还有，记得导入

import pandas as pd
import xlsxwriter

df = pd.DataFrame({'Data': [10, 20, 30, 20, 15, 30, 45]})
writer = pd.ExcelWriter('pandas_simple.xlsx', engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1')
writer.save()