Question

我使用pandas（来自this answer)，将DataFrame matplotlib绘制为表格。

现在我要设置给定行的底部边缘颜色，并且我有以下代码：

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

df = pd.DataFrame()
df['date'] = ['2016-04-01', '2016-04-02', '2016-04-03', '2016-04-04']
df['calories'] = [2200, 2100, 1500, 1800]
df['sleep hours'] = [2200, 2100, 1500, 1500]
df['gym'] = [True, False, False, True]

def render_mpl_table(data, col_width=3.0, row_height=0.625, font_size=14,
                     header_color='#40466e', row_colors=['#f1f1f2', 'w'], edge_color='w',
                     bbox=[0, 0, 1, 1], header_columns=0,
                     ax=None, **kwargs):
    if ax is None:
        size = (np.array(data.shape[::-1]) + np.array([0, 1])) * np.array([col_width, row_height])
        fig, ax = plt.subplots(figsize=size)
        ax.axis('off')

    mpl_table = ax.table(cellText=data.values, bbox=bbox, colLabels=data.columns, **kwargs)

    mpl_table.auto_set_font_size(False)
    mpl_table.set_fontsize(font_size)

    for k, cell in six.iteritems(mpl_table._cells):
        cell.set_edgecolor(edge_color)
        if k[0] == 0 or k[1] < header_columns:
            cell.set_text_props(weight='bold', color='w')
            cell.set_facecolor(header_color)
        else:
            cell.set_facecolor(row_colors[k[0]%len(row_colors) ])
    return ax

def get_table(ax):
    table = None
    for child in ax.get_children():
        if isinstance(child, matplotlib.table.Table):
            table = child
            return table
    return table

def set_row_edge_color(ax, row, color):
    table = get_table(ax)
    for k, cell in  six.iteritems(table._cells):
        if (k[0] == row):
            cell.set_edgecolor(color)

ax = render_mpl_table(df, header_columns=0, col_width=2.0)
set_row_edge_color(ax, 2, 'k')
plt.show()

我无法仅设置行底部的颜色，它的设置如下：

有没有办法像这样仅设置行底颜色？

还是有一种方法可以在图形/图中找到行并画一条水平线？

Answer 1

在matplotlib中，没有通用的方法可以在单元格（Rectangle s）的各个面上制作不同粗细或颜色的线条。对于问题，由于表格填满了轴的整个边界框，因此可以很容易地通过ax.axhline()（如@GAnderson评论）获得解决方案。

您首先需要将轴的数据范围设置为-1和表中的行数之间的范围。然后，您可以在选择的位置绘制axhline。

仅更改了两行（我用注释标记了），看来您可以完全摆脱get_table函数的作用。

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

df = pd.DataFrame()
df['date'] = ['2016-04-01', '2016-04-02', '2016-04-03', '2016-04-04']
df['calories'] = [2200, 2100, 1500, 1800]
df['sleep hours'] = [2200, 2100, 1500, 1500]
df['gym'] = [True, False, False, True]

def render_mpl_table(data, col_width=3.0, row_height=0.625, font_size=14,
                     header_color='#40466e', row_colors=['#f1f1f2', 'w'], edge_color='w',
                     bbox=[0, 0, 1, 1], header_columns=0,
                     ax=None, **kwargs):
    if ax is None:
        size = (np.array(data.shape[::-1]) + np.array([0, 1])) * np.array([col_width, row_height])
        fig, ax = plt.subplots(figsize=size)
        ax.axis('off')
    ax.axis([0,1,data.shape[0],-1])                ## <---------- Change here
    mpl_table = ax.table(cellText=data.values, bbox=bbox, colLabels=data.columns, **kwargs)

    mpl_table.auto_set_font_size(False)
    mpl_table.set_fontsize(font_size)

    for k, cell in six.iteritems(mpl_table._cells):
        cell.set_edgecolor(edge_color)
        if k[0] == 0 or k[1] < header_columns:
            cell.set_text_props(weight='bold', color='w')
            cell.set_facecolor(header_color)
        else:
            cell.set_facecolor(row_colors[k[0]%len(row_colors) ])
    return ax


def set_row_edge_color(ax, row, color):
    ax.axhline(y=row, color=color)                  ## <---------- Change here

ax = render_mpl_table(df, header_columns=0, col_width=2.0)
set_row_edge_color(ax, 2, 'k')
plt.show()

设置matplotlib表的行边缘颜色

1 个答案: