Python 如何在Matplotlib中旋转表格标题?

Python 如何在Matplotlib中旋转表格标题?,python,pandas,matplotlib,Python,Pandas,Matplotlib,目标是使用df中的Matplotlib创建表。然后,我希望标题(日期、卡路里、睡眠时间)旋转90度 最初,我认为这可以通过访问每个标签并使用label.set\u rotation(90)来实现。然而,它并没有如预期的那样实现 我可以知道如何解决这个问题吗 完整代码如下所示 import pandas as pd import numpy as np import matplotlib.pyplot as plt import six df = pd.DataFrame () df ['dat

目标是使用df中的Matplotlib创建表。然后,我希望标题(日期、卡路里、睡眠时间)旋转90度

最初,我认为这可以通过访问每个标签并使用
label.set\u rotation(90)
来实现。然而,它并没有如预期的那样实现

我可以知道如何解决这个问题吗

完整代码如下所示

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]

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

    for label in ax.get_xticklabels():
        # https://stackoverflow.com/a/43153984/6446053
        label.set_ha("right")
        label.set_rotation(90)

    return ax

    render_mpl_table (df, header_columns=0, col_width=2.0)
    
    plt.show ()
将熊猫作为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']=[真、假、假]
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(行颜色)])
对于ax.Getxticklabels()中的标签:
# https://stackoverflow.com/a/43153984/6446053
标签。设置_ha(“右”)
标签。设置旋转(90)
返回斧头
渲染mpl表格(df,标题列=0,列宽度=2.0)
plt.show()

提前感谢。

您可以在标题行中设置文本属性
旋转
,如下所示:

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]

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, header_height=2.0,
                     ax=None, **kwargs):
    if ax is None:
        size = (np.array(data.shape[::-1]) + np.array([0, 1])) * np.array([col_width, row_height])
        size[1] += header_height - 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)
            cell.set_text_props(rotation='vertical')
            cell.set_height(header_height)
        else:
            cell.set_facecolor(row_colors[k[0]%len(row_colors) ])
            cell.set_height(row_height)


    return ax

render_mpl_table (df, header_columns=0, col_width=2.0, header_height=2.0)
    
plt.show ()
将熊猫作为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']=[真、假、假]
def呈现mpl表格(数据,列宽=3.0,行高=0.625,字体大小=14,
标题颜色='#40466e',行颜色=['#f1f1f2',w'],边颜色='w',
bbox=[0,0,1,1],标题列=0,标题高度=2.0,
ax=无,**kwargs):
如果ax为无:
size=(np.array(data.shape[::-1])+np.array([0,1])*np.array([col\u-width,row\u-height])
大小[1]+=标题高度-行高度
图,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')
单元格设置面颜色(标题颜色)
cell.set_text_props(旋转=垂直)
单元格设置高度(标题高度)
其他:
单元格.设置面颜色(行颜色[k[0]%len(行颜色)])
单元格设置高度(行高度)
返回斧头
渲染mpl表格(df,标题列=0,列宽=2.0,标题高度=2.0)
plt.show()


请注意,我在示例中手动调整了行高。

不幸的是,表列名无法旋转。对于
ax中的标签。get_xticklabels():打印(标签);文本(0,0,,)
请参阅。