Как отображать таблицы под каждым подграфиком, используя метки осей в качестве заголовков столбцов
Я пытаюсь отобразить таблицу данных под каждым из двух подграфиков на рисунке, который я построил с помощью функции pandas plot. У меня есть графики, которые выглядят так, как я хочу, и я могу заставить одну из двух таблиц отображаться под одним из двух подзаголовков, но это нечитаемо (и только одна из двух, которые я хочу отобразить).
Моя цель состоит в том, чтобы каждый график и таблица выглядели примерно так: 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()
Как видите, это не дает ничего особенно красивого или даже читабельного. Что нужно изменить в этом коде, чтобы он работал как в примере по ссылке?
Ответы 1
What needs to be changed in this code to make it work as in the example in the link?
Набор данных в примере сильно отличается от того, что указано в вашем вопросе. Кроме того, есть два подзаголовка и таблицы. Для удобочитаемости вы можете изменить код, чтобы увеличить размеры фигур, освободить место для таблиц, показать таблицу под каждым подграфиком, скрыть метки 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()
Выход
Таким образом, он автоматически знает, что заголовки столбцов следует устанавливать в качестве значений оси? Потому что это никогда и нигде не указывается.
Метки столбцов указываются со следующим аргументом: colLabels = df['Assessment']
В настоящее время ваша ось x находится в диапазоне от 0 до 3. Но для того, чтобы галочки были выровнены с таблицей, она должна варьироваться от -0,5 до 3,5 или, в более общем случае, до
n+0.5, гдеn— это количество уникальных меток в столбце x.