[英]Setting row edge color of matplotlib table
我有一個使用matplotlib
作為表格繪制的pandas
DataFrame
(來自這個答案) 。
現在我想設置給定行的底邊顏色,我有這個代碼:
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()
有沒有辦法只設置這樣的行底顏色?
或者有沒有辦法在圖/情節中找到行並繪制一條水平線?
在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()
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.