Matplotlib表格式化列宽

Question

我希望格式化表格的一列，但是当遍历行时，它的宽度似乎在每次迭代后都会改变。

源代码

def p_create_table(self, events, dates, rows, columns, portfolio):
    """
    :param events: Dict - {Date:Event} where Event is a String identifying
    what event happened
    :param dates: List - Dates of events
    :param rows: Int - number of Dates (rows) to create for the table
    :param columns: List - Column headers
    :param portfolio: Dataframe - Portfolio with calculated totals and returns
    :return:
    """
    cell_text = self.p_create_cell_text(events, dates, portfolio)
    cell_text.pop(0)
    row_labels = self.p_create_row_labels(rows)
    row_labels.pop(len(row_labels) - 1)
    colors = self.p_set_table_colors(row_labels)

    table = plt.table(cellText=cell_text, cellColours=colors[0],
              rowColours=colors[1], rowLabels=row_labels,
              colColours=colors[2], colLabels=columns,
              bbox=[0.0, -1.3, 1.0, 1.0], cellLoc='center')

    table.auto_set_font_size(False)
    table.set_fontsize(9)
    table.scale(2, 2)

    cell_dict = table.get_celld()
    for i in range(13):
        cell_dict[(i,1)].set_width(0.3)

下图是调整大小之前的表格图片。table.set_fontsize(9)在执行行后拍摄快照。我想调整第二列的大小Event。

格式化之前

不幸的是，经过以下迭代：

for i in range(13):
    cell_dict[(i,1)].set_width(0.3)

看起来像单元格宽度增加，结果是这样的：

格式化后

对于为什么会发生这种情况的任何建议，或者调整宽度的替代解决方案，将不胜感激！

Answer 1

除了@jma 的回答，我发现 table 方法.auto_set_column_widths(col=<list of col indices>)非常有用，尤其是与.auto_set_font_size(False). 使用示例如下：

from matplotlib import pyplot as plt
import pandas as pd
# Create some example data
data = [{"Movie": "Happy Gilmore", "Lead Actor": "Adam Sandler" , "Year": "1996", 
              "Plot": "An ice hockey star takes up golfing.", 
              "Quotes": "\"Just give it a little tappy. Tap tap taparoo.\""}]
dff = pd.DataFrame(data)
fig, ax = plt.subplots(3,1, figsize=(10,4))

tab0 = ax[0].table(cellText=dff.values, colLabels=dff.columns, loc='center', cellLoc='center')
ax[0].set_title("Default")
tab1 = ax[1].table(cellText=dff.values, colLabels=dff.columns, loc='center', cellLoc='center')
ax[1].set_title("Font AutoSize off")
tab2 = ax[2].table(cellText=dff.values, colLabels=dff.columns, loc='center', cellLoc='center')
ax[2].set_title("Column Width Auto Set, Font AutoSize off")
[a.axis("off") for a in ax]
[t.auto_set_font_size(False) for t in [tab1, tab2]]
[t.set_fontsize(8) for t in [tab1, tab2]]

tab2.auto_set_column_width(col=list(range(len(dff.columns)))) # Provide integer list of columns to adjust
plt.show()

Answer 2

要设置列的宽度，请在表函数中使用colWidths。它接受每一列的宽度列表-它们都可以相同或不同。

table = plt.table(cellText=cell_text, cellColours=colors[0],
          rowColours=colors[1], rowLabels=row_labels,
          colColours=colors[2], colLabels=columns,
          colWidths=[0.3 for x in columns],
          bbox=[0.0, -1.3, 1.0, 1.0], cellLoc='center')