Python 是否设置matplotlib表格的单元格颜色并另存为图形?

Python 是否设置matplotlib表格的单元格颜色并另存为图形?,python,pandas,Python,Pandas,我遵守这条规则!为了将一个表保存为图像,我有一些功能,比如在单元格中检查值,然后为单元格设置颜色,但是我添加了一些代码样式图,它不起作用 import pandas as pd import numpy as np import matplotlib.pyplot as plt import six ......... df = pd.DataFrame() df['date'] = ['2016-04-01', '2016-04-02', '2016-04-03'] df['calori

我遵守这条规则!为了将一个表保存为图像,我有一些功能,比如在单元格中检查值,然后为单元格设置颜色,但是我添加了一些代码样式图,它不起作用

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



df = pd.DataFrame()
df['date'] = ['2016-04-01', '2016-04-02', '2016-04-03']
df['calories'] = [2200, 2100, 1500]
df['sleep hours'] = [2200, 2100, 1500]
df['gym'] = [True, False, False]
df.style.applymap(color)
render_mpl_table(df, header_columns=0, col_width=2.0)

.......
def color(val):
    if val < datetime.now():
        color = 'green'
    elif val > datetime.now():
        color = 'yellow'
    elif val > (datetime.now() + timedelta(days=60)):
        color = 'red'
    return 'background-color: %s' % color

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
将熊猫作为pd导入
将numpy作为np导入
将matplotlib.pyplot作为plt导入
进口六
.........
df=pd.DataFrame()
df['date']=['2016-04-01','2016-04-02','2016-04-03']
df['卡路里]=[22002100,1500]
df[‘睡眠时间’]=[220021001500]
df['gym']=[真、假、假]
df.style.applymap(颜色)
渲染mpl表格(df,标题列=0,列宽度=2.0)
.......
def颜色(val):
如果valdatetime.now():
颜色=‘黄色’
elif val>(datetime.now()+timedelta(天=60)):
颜色=‘红色’
返回“背景颜色:%s”%s
def呈现mpl表格(数据,列宽=3.0,行高=0.625,字体大小=14,
标题颜色='#40466e',行颜色=['#f1f1f2',w'],边颜色='w',
bbox=[0,0,1,1],标题\列=0,
ax=无,**kwargs):
如果ax为无:
size=(np.array(data.shape[::-1])+np.array([0,1])*np.array([col\u-width,row\u-height])
图,ax=plt.子批次(图尺寸=尺寸)
ax.轴(“关闭”)
mpl_table=ax.table(cellText=data.values,bbox=bbox,colLabels=data.columns,**kwargs)
mpl表格。自动设置字体大小(错误)
mpl_表格。设置字体大小(字体大小)
对于k,六个.iteritems中的单元格(mpl_表._单元格):
单元格。设置边缘颜色(边缘颜色)
如果k[0]==0或k[1]<标题列:
cell.set_text_props(weight='bold',color='w')
单元格设置面颜色(标题颜色)
其他:
单元格.设置面颜色(行颜色[k[0]%len(行颜色)])
返回斧头
我将样式放在dataframe下,然后进行渲染

我的解决方案是为单元格颜色创建新的数据框,使用表中选中的颜色值创建数据框,然后添加到
cellcolors
我的代码块:

#table datas
df = pd.DataFrame()
members = [x.display_name[:10] for x in message.server.members]
arrayDataFrame(members, 'Players', df)
arrayDataFrame(Data=members, df=df)
#table Colors
cf = pd.DataFrame()
arrayColorFrame(members, "Players", cf)
arrayColorFrame(Data=members, df=cf)
#render
plt = render_mpl_table(df, header_columns=0, col_width=2.0, CellCol= cf)

>>>>>>>>>>>>>>
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='B',
                 bbox=[0, 0, 1, 1], header_columns=0,
                 ax=None, CellCol=None, **kwargs):
>>>>>>>>>>>>>>>>>>>>>
>>>>>>>>>>>>>>>>>>>>>
      mpl_table = ax.table(cellText=data.values, bbox=bbox, colLabels=data.columns,
            cellColours=CellCol.values, **kwargs)


def arrayDataFrame(Data, Lable='No', df=None):
    if df is None:
        df = pd.DataFrame()
    leng = len(Data)
    array = list()
    if Lable == 'No':
        for index in range(leng):
            array.append(int(index + 1))
    else:
        for val in Data:
            array.append(val)
    df[Lable] = array
    return df

def arrayColorFrame(Data, Lable='No', df=None):
    if df is None:
        df = pd.DataFrame()
    leng = len(Data)
    array = list()
    if Lable == 'No':
        for index in range(leng):
            val = index + 1
            if val < 10:
                color = 'green'
            elif val < 20:
                color = 'yellow'
            elif val < 40:
                color = 'red'
            else:
                color = 'w'
            array.append(color)
    else:
        for val in Data:
            if "nat" in val :
                color = 'blue'
            else:
                color='w'
            array.append(color)
    df[Lable] = array
    return df
#表格数据
df=pd.DataFrame()
members=[x.display_name[:10]表示message.server.members中的x]
arrayDataFrame(成员,“玩家”,df)
arrayDataFrame(数据=成员,df=df)
#桌子颜色
cf=pd.DataFrame()
arrayColorFrame(成员,“玩家”,cf)
arrayColorFrame(数据=成员,df=cf)
#渲染
plt=渲染mpl表格(df,标题列=0,列宽度=2.0,单元格列=cf)
>>>>>>>>>>>>>>
def呈现mpl表格(数据,列宽=3.0,行高=0.625,字体大小=14,
标题颜色='#40466e',行颜色=['#f1f1f2',W'],边颜色='B',
bbox=[0,0,1,1],标题\列=0,
ax=无,CellCol=无,**kwargs):
>>>>>>>>>>>>>>>>>>>>>
>>>>>>>>>>>>>>>>>>>>>
mpl_table=ax.table(cellText=data.values,bbox=bbox,colLabels=data.columns,
cellcolors=CellCol.values,**kwargs)
def arrayDataFrame(数据,标签='No',df=None):
如果df为无:
df=pd.DataFrame()
长度=长度(数据)
数组=列表()
如果标签==“否”:
对于范围内的索引(长度):
array.append(int(索引+1))
其他:
对于数据中的val:
array.append(val)
df[Lable]=数组
返回df
def arrayColorFrame(数据,标签='No',df=None):
如果df为无:
df=pd.DataFrame()
长度=长度(数据)
数组=列表()
如果标签==“否”:
对于范围内的索引(长度):
val=索引+1
如果val<10:
颜色=‘绿色’
elif val<20:
颜色=‘黄色’
elif val<40:
颜色=‘红色’
其他:
颜色='w'
array.append(颜色)
其他:
对于数据中的val:
如果val中的“nat”:
颜色=‘蓝色’
其他:
color='w'
array.append(颜色)
df[Lable]=数组
返回df