python - 如何在 matplotlib 表上显示数据帧索引名称？

Question

我有一个数据框，我想通过 matplotlib 在 PDF 文件中显示为表格。

我的代码看起来像这样(我绕过了我之前做的事情，因为数据与我的问题无关，我想):

table = ax_table.table(cellText=str_df.values, rowLabels=str_df.index, cellLoc='center',
                       colColours=['gainsboro'] * len(table_col), colLabels=table_col, loc='center',
                       colWidths= [0.12]*(len(table_col)), cellColours = img.to_rgba(df_for_table.values))

我的数据框 (str_df) 索引名称是“Area”，我想通过 matplotlib 表显示它而不是有空格(用红色圈出):

有人知道这是否可行吗？

谢谢。

Answer 1

您可以在相应位置添加一个单元格，并使用 dataframe.index.name 作为其文本。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame(np.random.randint(1,50, size=(8,4)), columns=list("ABCD"))
df.index.name = "Name"


fig, ax = plt.subplots()
table = ax.table(cellText=df.values, rowLabels=df.index, cellLoc='center',
                 colColours=['gainsboro'] * len(df), colLabels=df.columns, loc='center',
                 colWidths= [0.12]*(len(df.columns)))
w, h = table[0,1].get_width(), table[0,1].get_height()
table.add_cell(0, -1, w,h, text=df.index.name)
plt.show()