matplotlib 如何在pyplot.table中旋转列标题?

hgqdbh6s  于 2022-11-15  发布在  其他
关注(0)|答案(3)|浏览(123)

我在matplotlib中创建了一个表,但是表头是很长的字符串,表值是只有几位数的数字。这给我留下了两个糟糕的选择:我的表格比需要的宽很多,或者我的标题重叠了。为了解决这个问题,我想旋转表格标题(可能高达90度)。换句话说,我想在python中执行this
下面是我的简化代码:

import matplotlib, numpy
import matplotlib.pyplot as plt

data=numpy.array([[1, 2],[3,4]])
headings=['long heading 1', 'long heading 2']
fig=plt.figure(figsize=(5,2))
ax=fig.add_subplot(111, frameon=False, xticks=[], yticks=[])

the_table = plt.table(cellText=data, rowLabels=headings, colLabels=headings, colWidths=[0.3]*data.shape[1], loc='center') #0.3 for second image, 0.03 for first
#the_table.auto_set_font_size(False) #comment out for second image
#the_table.set_fontsize(10) #comment out for second image
the_table.scale(1, 1.6)
plt.show()

这会产生压扁的图像或超宽的图像(如下所示)。在我的实际代码中,表格是~30 x 30,所以单元格不能太宽。有人知道如何旋转列标题来解决这个间距问题吗?x1c 0d1xx 1c 1d 1x

mkshixfv

mkshixfv1#

我想出来了。它不漂亮,但很好用。我为每列添加了两个注解-文本和一条线将其与下一列标题分隔开。我必须定义一些适用于表格和花哨标签(宽度、高度、列宽)的参数,以及一些使花哨标签正确排列的参数。这个解决方案在我的30 x30表格上运行良好。

import matplotlib, numpy
import matplotlib.pyplot as plt

width=5
height=3
col_width=.075

data=numpy.array([[1, 2,5],[3,4,7],[7,9,5]])
headings=['long heading 1', 'long heading 2', 'longish 3']
fig=plt.figure(figsize=(width,height))
ax=fig.add_subplot(111, frameon=False, xticks=[], yticks=[])

the_table = plt.table(cellText=data, rowLabels=headings, 
    colWidths=[col_width]*data.shape[1], loc='center') #remove colLabels
the_table.auto_set_font_size(False) 
the_table.set_fontsize(10) 
the_table.scale(1, 1.6)

#custom heading titles - new portion
hoffset=0.42 #find this number from trial and error
voffset=0.66 #find this number from trial and error
line_fac=0.98 #controls the length of the dividing line
count=0
for string in headings:
    ax.annotate('  '+string, xy=(hoffset+count*col_width,voffset),
        xycoords='axes fraction', ha='left', va='bottom', 
        rotation=45, size=10)

    #add a dividing line
    ax.annotate('', xy=(hoffset+(count+0.5)*col_width,voffset), 
        xytext=(hoffset+(count+0.5)*col_width+line_fac/width,voffset+line_fac/height),
        xycoords='axes fraction', arrowprops={'arrowstyle':'-'})

    count+=1

plt.show()

m528fe3b

m528fe3b2#

我找到了另一个解决办法:

for cell in table._cells:
    if cell[0] ==0:
        table._cells[cell].get_text().set_rotation(90)

第一个循环用于遍历所有单元格,第二个循环用于选取第一行/标题。

if cell[1] =- -1

这将选择第一列,您可能也想旋转它。
然后您可以旋转单元格文本,例如90 °。

l2osamch

l2osamch3#

以下内容对我很有效。
创建表

fig = plt.figure( figsize=(pageWidthInInches, pageHeightInInches) )
    panel = plotUtils.createPanelSameSizeAsFig(fig)

    tablePanel = panel.table(
                        cellText=cellText
                        # ,rowLabels=rowLabels
                        # ,colLabels=colLabels
                        #,loc='center' # center table in panel, title is in center
                        #,loc='bottom' # center table in panel does not work well
                        ,loc='best'
                        #,rowColours=aedwip,
                        #,colColourslist=aedwip
                        ,cellColours= cellColors
                       )

    # get rid of bar chart axis and box 
    panel.get_xaxis().set_visible(False)
    panel.get_yaxis().set_visible(False)

    tablePanel.scale(1, 1.5)

    plt.box(on=None)
    panel.set_title(title)

现在添加列标题。注意我的表没有行标签,您可能需要调整startX的位置

tcell = table._cells[(0, 0)]
cellWidth = tcell.get_width()
startX  = tcell.get_x() - cellWidth
y = 0.99 #0.98 #0.975 #0.96 #1

headings = sampleDF.columns
for i in range(len(headings)): 
    heading = headings[i]
    x = startX + i * cellWidth
    
    panel.text(x, y, heading, horizontalalignment="left", 
              verticalalignment="baseline", rotation=45, fontsize=4)

相关问题