Matplotlib-在单元格中创建一个包含线图的表格?

时间:2017-12-12 18:47:38

标签: python pandas matplotlib plot

我有历史棒球数据,我试图在一个简单的matplotlib图中可视化在1个子图中我希望有一个表格,显示过去一年的平均统计数据,每个统计数据的折线图,然后是最终得分,独立计算。

enter image description here

我知道matplotlib有一个表函数,所以创建一个5x3表很简单,但是可以在表中插入一个值吗?如果没有,对我该做什么有什么建议吗?我想我可以创建几个子图,但格式化将是古怪的,而不是非常动态。感谢帮助。

使用@ TheImportanceofBeingErnest的代码,我遇到了matplotlib的错误,当我使用gridspec时,我无法看到x轴:

fig = plt.figure(figsize=(8.5,11))

gs_row1left = gridspec.GridSpec(1,1)
gs_row1right = gridspec.GridSpec(1,1)

summaryplot2subplot(fig, gs_row1left[0], data, col1, col2, finalsc)
ax.axis('off')

ax2 = fig.add_subplot(gs_row1right[0, 0])
df = pd.DataFrame({'year':['2001-01','2002-01','2003-01','2004-01','2005-01'], 'value':[100,200,300,400,500]})

barax = ax2.twinx()
df['value1']= df['value']*0.4

df['value2'] = df['value']*0.6# Let them be strings!



df.plot(x = ['year'], y = ['value'], kind = 'line', ax = ax2)

df.plot(x = ['year'], y= ['value1','value2'], kind = 'bar', ax = barax)



gs_row1left.update(left = 0.05, right = 0.48)
gs_row1right.update(left = 0.55, right = 0.98)
plt.show()

enter image description here

1 个答案:

答案 0 :(得分:3)

无法将图表插入到matplotlib表中。但是,子图网格允许创建类似表格的行为。

import matplotlib.pyplot as plt
import numpy as np

data = np.random.rand(100,4)
col1 = ["WAR", "ERA", "IP", "WHIP", "Final\nScore"]
col2 = [0.23,1.60,0.28,0.02,0.38]
col2colors = ["red", "g", "r", "r", "r"]
finalsc = "D+"

fig, axes = plt.subplots(ncols=3, nrows=5, figsize=(4,2.6),
                         gridspec_kw={"width_ratios":[1,0.5,2]})
fig.subplots_adjust(0.05,0.05,0.95,0.95, wspace=0.05, hspace=0)

for ax in axes.flatten():
    ax.tick_params(labelbottom=0, labelleft=0, bottom=0, top=0, left=0, right=0)
    ax.ticklabel_format(useOffset=False, style="plain")
    for _,s in ax.spines.items():
        s.set_visible(False)
border = fig.add_subplot(111)
border.tick_params(labelbottom=0, labelleft=0, bottom=0, top=0, left=0, right=0)
border.set_facecolor("None")

text_kw = dict(ha="center", va="bottom", size=13)
for i,ax in enumerate(axes[:,0]):
    ax.text(0.5, 0.05, col1[i], transform=ax.transAxes, **text_kw)
for i,ax in enumerate(axes[:,1]):
    ax.text(0.5, 0.05, "{:.2f}".format(col2[i]),transform=ax.transAxes, **text_kw)
    ax.set_facecolor(col2colors[i])
    ax.patch.set_color(col2colors[i])
axes[-1,-1].text(0.5, 0.05, finalsc,transform=axes[-1,-1].transAxes, **text_kw)

for i,ax in enumerate(axes[:-1,2]):
    ax.plot(data[:,i], color="green", linewidth=1)


plt.show()

enter image description here

要将几个这样的图形放入图中,您可以稍微改变一下,并创建一个包含多个子网格的gridspec。



import matplotlib.pyplot as plt
from matplotlib import gridspec
import numpy as np


def summaryplot2subplot(fig, gs, data, col1, col2, finalsc):
    col2colors = ["g" if col2[i] > 1 else "r" for i in range(len(col2)) ]
    sgs = gridspec.GridSpecFromSubplotSpec(5,3, subplot_spec=gs, wspace=0.05, hspace=0,
                                           width_ratios=[0.9,0.7,2])
    axes = []
    for n in range(5):
        for m in range(3):
            axes.append(fig.add_subplot(sgs[n,m]))
    axes = np.array(axes).reshape(5,3)
    for ax in axes.flatten():
        ax.tick_params(labelbottom=0, labelleft=0, bottom=0, top=0, left=0, right=0)
        ax.ticklabel_format(useOffset=False, style="plain")
        for _,s in ax.spines.items():
            s.set_visible(False)
    border = fig.add_subplot(gs)
    border.tick_params(labelbottom=0, labelleft=0, bottom=0, top=0, left=0, right=0)
    border.set_facecolor("None")
    
    text_kw = dict(ha="center", va="bottom", size=11)
    for i,ax in enumerate(axes[:,0]):
        ax.text(0.5, 0.05, col1[i], transform=ax.transAxes, **text_kw)
    for i,ax in enumerate(axes[:,1]):
        ax.text(0.5, 0.05, "{:.2f}".format(col2[i]),transform=ax.transAxes, **text_kw)
        ax.set_facecolor(col2colors[i])
        ax.patch.set_color(col2colors[i])
    axes[-1,-1].text(0.5, 0.05, finalsc,transform=axes[-1,-1].transAxes, **text_kw)
    
    for i,ax in enumerate(axes[:-1,2]):
        ax.plot(data[:,i], color=col2colors[i], linewidth=1)


fig = plt.figure(figsize=(8,6))
gs = gridspec.GridSpec(2,2)


col1 = ["WAR", "ERA", "IP", "WHIP", "Final\nScore"]
finalsc = "D+"

for i in range(4):
    data = np.random.rand(100,4)
    col2 = np.random.rand(5)*2
    summaryplot2subplot(fig, gs[i], data, col1, col2, finalsc)

plt.show()




enter image description here