Как отобразить таблицы под каждым субплотом, используя метки осей в качестве заголовков столбцов

Question

Я пытаюсь отобразить таблицу данных под каждым из двух вспомогательных участков на рисунке, которые я построил с помощью функции plot pandas.У меня есть графики, чтобы они выглядели так, как я хочу, и могу заставить одну из двух таблиц отображаться под одним из двух вспомогательных участков, но это нечитабельно (и я хочу отобразить только одну из двух).

Моя цель - чтобы каждый график и таблица выглядели примерно так: https://matplotlib.org/gallery/misc/table_demo.html#sphx-glr-gallery-misc-table-demo-py, где оси выступают в качестве заголовков столбцов, а данные находятся непосредственно под каждым графиком.К сожалению, моя фигура не выходит так.Ниже приведен мой код, который при запуске должен выдавать именно то, что я вижу.

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

# Set up df
rowcol = {'ID':[101,101,101,101], 'Assessment': ['Read-Write-1','Math-1', 'Read-Write-Math-3', 'Read-Write-Math-4'],
          'Math': [np.nan,4,3,3], 'MScore': [np.nan, 636.5, 577.2, 545.4],
          'RW': [3, np.nan, 3, 3], 'RWScore': [559.7, np.nan, 621.6, 563.7]}
df = pd.DataFrame(data = rowcol)
df = df.interpolate()

# Set up subplots
fig, axes = plt.subplots(nrows=1, ncols=2, sharex=True)
df.plot(x='Assessment', y=['MScore', 'RWScore'], use_index = True, 
        grid = True, style=['+r-.', 'xb--'], legend=['MScore','RWScore'], 
        ax=axes[0], xticks=[0,1,2,3])
df.plot(x='Assessment', y=['Math', 'RW'], use_index = True, 
        grid = True, style=['+g-.', 'xc--'], legend=['M','RW'], 
        xticks=[0,1,2,3], ylim=[0,4], yticks=[1,2,3,4], ax=axes[1])

# Add labels, titles, and legend
axes[0].set_title(df['ID'][0])
axes[1].set_title(df['ID'][0])
plt.xlabel('Assessment')
axes[0].set_ylabel('Score')
axes[1].set_ylabel('Performance Level')
plt.legend(loc='best')

# Add data tables
table1 = plt.table(cellText = [df.MScore, df.RWScore],
                   rowLabels = ['MScore', 'RWScore'],
                   rowColours = ['r','b'], loc='bottom',
                   colLabels = df['Assessment'])
table2 = plt.table(cellText = [df.Math, df.RW],
                   rowLabels = ['Math', 'RW'],
                   rowColours = ['g','c'], loc='bottom',
                   colLabels = df['Assessment'])

# Show plot
plt.show()

Как видите, это не дает ничего особенно красивого или даже читабельного.Что нужно изменить в этом коде, чтобы он работал, как в примере в ссылке?

Answer 1

Что нужно изменить в этом коде, чтобы он работал, как в примере в ссылке?

Набор данных в примере сильно отличается от того, который был задан в вашем вопросе.Кроме того, есть два вспомогательных участка и таблицы.Для удобства чтения вы можете изменить код, чтобы увеличить размеры рисунков, освободить место для таблиц, отобразить таблицу под каждым вложенным участком, скрыть метки xticklabels и оси x.

rowcol = {'ID':[101,101,101,101], 'Assessment': ['Read-Write-1','Math-1', 'Read-Write-Math-3', 'Read-Write-Math-4'],
          'Math': [np.nan,4,3,3], 'MScore': [np.nan, 636.5, 577.2, 545.4],
          'RW': [3, np.nan, 3, 3], 'RWScore': [559.7, np.nan, 621.6, 563.7]}
df = pd.DataFrame(data = rowcol)
df = df.interpolate()
# print(df)
# Set up subplots
fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(30, 10)) #specify size of subplots
df.plot(x='Assessment', y=['MScore', 'RWScore'], use_index = True, 
        grid = True, style=['+r-.', 'xb--'], legend=['MScore','RWScore'], 
        ax=axes[0], xticks=[0,1,2,3])
df.plot(x='Assessment', y=['Math', 'RW'], use_index = True, 
        grid = True, style=['+g-.', 'xc--'], legend=['M','RW'], 
        xticks=[0,1,2,3], ylim=[0,4], yticks=[1,2,3,4], ax=axes[1])


# Add labels, titles, and legend
plt.subplots_adjust(left=0.3, bottom=0.2, wspace = 0.3)
axes[0].set_title(df['ID'][0])
axes[1].set_title(df['ID'][0])
axes[0].set_ylabel('Score')
axes[1].set_ylabel('Performance Level')
#set visibility of x-axis and y-axis, xticklabels and yticklabels
axes[0].xaxis.set_ticklabels([])
axes[1].xaxis.set_ticklabels([])
axes[0].get_xaxis().set_visible(False)
axes[1].get_xaxis().set_visible(False)
plt.legend(loc='best')


# Add data tables for each subplot
table1 = axes[0].table(cellText = [df.MScore, df.RWScore],
                   rowLabels = ['MScore', 'RWScore'],
                   rowColours = ['r','b'], loc='bottom',
                   colLabels = df['Assessment'])
table2 = axes[1].table(cellText = [df.Math, df.RW],
                   rowLabels = ['Math', 'RW'],
                   rowColours = ['g','c'], loc='bottom',
                   colLabels = df['Assessment'], fontsize=15)

# Show plot
plt.show()

Вывод