设置 matplotlib 表的行边缘颜色

我pandas DataFrame使用matplotlib(from this answer)绘制了一个表格。


现在我想设置给定行的底部边缘颜色,我有这个代码:


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()

我无法只设置行底部的颜色,它设置如下:

http://img4.mukewang.com/60e3c22500016bf806200237.jpg

有没有办法像这样只设置行底颜色?

http://img1.mukewang.com/60e3c22f0001b82106190230.jpg

或者有没有办法在图形/图中定位行并绘制一条水平线?


倚天杖
浏览 154回答 1
1回答

莫回无

Rectangle在 matplotlib 中,没有通用的方法可以在单元格 ( s) 的各个侧面制作不同粗细或颜色的线条。在这个问题的情况下,ax.axhline()由于表格填充了轴的整个边界框,因此很容易通过(如@GAnderson 所评论)获得解决方案。您首先需要将轴的数据范围设置为 -1 和表中的行数之间的范围。然后你可以axhline在选择的位置绘制一个。只有两行更改(我用注释标记),看来您可以完全摆脱该get_table功能。import pandas as pdimport numpy as npimport matplotlib.pyplot as pltimport matplotlibimport sixdf = 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,&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp;header_color='#40466e', row_colors=['#f1f1f2', 'w'], edge_color='w',&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp;bbox=[0, 0, 1, 1], header_columns=0,&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp;ax=None, **kwargs):&nbsp; &nbsp; if ax is None:&nbsp; &nbsp; &nbsp; &nbsp; size = (np.array(data.shape[::-1]) + np.array([0, 1])) * np.array([col_width, row_height])&nbsp; &nbsp; &nbsp; &nbsp; fig, ax = plt.subplots(figsize=size)&nbsp; &nbsp; &nbsp; &nbsp; ax.axis('off')&nbsp; &nbsp; ax.axis([0,1,data.shape[0],-1])&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; ## <---------- Change here&nbsp; &nbsp; mpl_table = ax.table(cellText=data.values, bbox=bbox, colLabels=data.columns, **kwargs)&nbsp; &nbsp; mpl_table.auto_set_font_size(False)&nbsp; &nbsp; mpl_table.set_fontsize(font_size)&nbsp; &nbsp; for k, cell in six.iteritems(mpl_table._cells):&nbsp; &nbsp; &nbsp; &nbsp; cell.set_edgecolor(edge_color)&nbsp; &nbsp; &nbsp; &nbsp; if k[0] == 0 or k[1] < header_columns:&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; cell.set_text_props(weight='bold', color='w')&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; cell.set_facecolor(header_color)&nbsp; &nbsp; &nbsp; &nbsp; else:&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; cell.set_facecolor(row_colors[k[0]%len(row_colors) ])&nbsp; &nbsp; return axdef set_row_edge_color(ax, row, color):&nbsp; &nbsp; ax.axhline(y=row, color=color)&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; ## <---------- Change hereax = render_mpl_table(df, header_columns=0, col_width=2.0)set_row_edge_color(ax, 2, 'k')plt.show()
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python