Python 在写入 PDF 文件(PdfPages)时突出显示具有 NaN(matplotlib)的数据框

发布于03月25日

我想做两件事:

用红色突出显示数据帧的‘NaN’值.
将数据帧添加到PDF文件.

我能够在PDF页面中成功地显示dataframe个值，但是NaN个值不会在PDF中反映为红色.

我try 了以下代码:

    df.style.highlight_null('red') 

    with PdfPages('stale_curve_report.pdf') as pdf:
      fig, ax = plt.subplots()
      ax.axis('off')
      ax.table(cellText=df.values, colLabels=df.columns, rowLabels=df.index, loc='center',colWidths=[0.12] * 15)
      pdf.savefig(fig)
      plt.close(fig)

我还试过其他几种方法:

sns.heatmap(df.isna(), cmap=['red', 'white', 'white'])

我想，我需要ax.table中的一个选项来突出显示数据帧.

推荐答案

这可以通过在ax.table函数中创建cellColors种 colored颜色的列表来实现.为此，我们创建了一个逻辑数据帧color = df.isna()，用所需的 colored颜色替换接收到的True和False，并将其转换为列表.示例:

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages

df = pd.DataFrame(np.random.random((10, 3)), columns=("col 1", "col 2", "col 3"))
df.at[1, 'col 2'] = np.NaN
df.at[8, 'col 1'] = np.NaN
df.loc[2:4, ['col 3']] = np.NaN

color = df.isna()
color.replace({True: 'red', False: 'white'}, inplace=True)
list_color = color.values.tolist()

fig, ax = plt.subplots(figsize=(12, 4))
ax.axis('tight')
ax.axis('off')

the_table = ax.table(cellText=df.values, colLabels=df.columns, loc='center', cellColours=list_color)

pp = PdfPages("foo.pdf")
pp.savefig(fig)
pp.close()