类型错误：在没有 'nan_inf_to_errors' Workbook() 选项的 write_number() 中不支持 NAN/INF

Question

我想将尺寸为 (3960, 225) 的X (ndarray) 保存在 Excel 文件 (.xlsx) 中。在X中，我有一些缺失值（nan）。我为它编写了一个代码。但是，我收到错误。

这是代码：

workbook = xlsxwriter.Workbook('arrays.xlsx')
    worksheet = workbook.add_worksheet()
    row = 0
    for col, data in enumerate(X):
        worksheet.write_column(row, col, data)
    workbook.close()
     df = pd.DataFrame(X)
## save to xlsx file
     filepath = 'my_excel_file.xlsx'
      df.to_excel(filepath, index=False)

这是回溯：

Traceback (most recent call last):
  File "C:\Program Files\JetBrains\PyCharm 2020.2.1\plugins\python\helpers\pydev\pydevd.py", line 1448, in _exec
    pydev_imports.execfile(file, globals, locals)  # execute the script
  File "C:\Program Files\JetBrains\PyCharm 2020.2.1\plugins\python\helpers\pydev\_pydev_imps\_pydev_execfile.py", line 18, in execfile
    exec(compile(contents+"\n", file, 'exec'), glob, loc)
  File "C:/Users/Nafees Ahmed/PycharmProjects/Extra_Sensory_Experimetns/main.py", line 475, in <module>
    worksheet.write_column(row, col, data)
  File "C:\Users\Nafees Ahmed\AppData\Local\Programs\Python\Python38\lib\site-packages\xlsxwriter\worksheet.py", line 69, in cell_wrapper
    return method(self, *args, **kwargs)
  File "C:\Users\Nafees Ahmed\AppData\Local\Programs\Python\Python38\lib\site-packages\xlsxwriter\worksheet.py", line 1164, in write_column
    error = self._write(row, col, token, cell_format)
  File "C:\Users\Nafees Ahmed\AppData\Local\Programs\Python\Python38\lib\site-packages\xlsxwriter\worksheet.py", line 481, in _write
    return self._write_number(row, col, *args)
  File "C:\Users\Nafees Ahmed\AppData\Local\Programs\Python\Python38\lib\site-packages\xlsxwriter\worksheet.py", line 589, in _write_number
    raise TypeError(
TypeError: NAN/INF not supported in write_number() without 'nan_inf_to_errors' Workbook() option
 

也许，这是由于 nan（缺失）值造成的。有没有简单的方法来处理这个错误？

Answer 1

用零填充NaN值并不能解决问题，如果您想将NaN值保留为NaN，则应该跳过填充值，如下所示：

row = 0
for col, data in enumerate(X):
    try:
        worksheet.write_column(row, col, data)
    except:
        pass

Answer 2

您可以使用pandas 的fillna方法。

df.fillna(0)将全部替换NaN为0.0