добавить существующий сюжет в структуру осей / подпланов

Question

У меня есть df s, например:

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

df = pd.DataFrame({
  'Country': ["A", "B", "C", "D", "E", "F", "G"],
  'Answer declined': [0.000000, 0.000000, 0.000000, 0.000667, 0.000833, 0.000833, 0.000000],
  "Don't know": [0.003333, 0.000000, 0.000000, 0.001333, 0.001667, 0.000000, 0.000000],
  "No": [0.769167, 0.843333, 0.762000, 0.666000, 0.721667, 0.721667, 0.775833],
  "Yes": [0.227500, 0.156667, 0.238000, 0.332000, 0.275833, 0.277500, 0.224167]}, )
df.set_index("Country", inplace = True)

Поскольку у меня есть несколько таких df s, из которых я хочу создавать графики, я определил следующую функцию:

def bar_plot(plot_df):
    N = len(plot_df) # number of groups
    ind = np.arange(N) # x locations for the groups
    width = 0.35 # width of bars

    p_s = []
    p_s.append(plt.bar(ind, plot_df.iloc[:,0], width))
    for i in range(1,len(plot_df.columns)):
        p_s.append(plt.bar(ind, plot_df.iloc[:,i], width,
                           bottom=np.sum(plot_df.iloc[:,:i], axis=1)))

    plt.ylabel('[%]')
    plt.title('Responses by country')

    x_ticks_names = tuple([item for item in plot_df.index])

    plt.xticks(ind, x_ticks_names)
    plt.yticks(np.arange(0, 1.1, 0.1)) # ticks from, to, steps
    #if num_y_cats % 3 == 0: ncol = num_y_cats / 3
    #else: ncol = num_y_cats % 3
    ncol = 3
    plt.legend(p_s, plot_df.columns,
               bbox_to_anchor = (0.5, -0.25), # to the left; to the top
               loc = 'lower center',
               ncol = ncol,
               borderaxespad = 0)
    plt.show()
    plt.close()

вызов функции (bar_plot(df)) дает желаемый график.Тем не менее, я хочу манипулировать / точно настраивать графики и поэтому хочу встроить график в mpl figure s и axe s , но мне не удалось это сделать, так как я не могу понятькак заставить его работать со строками p_s = [] и p_s.append(...).

Может ли кто-нибудь мне помочь, куда пойдут fig = plt.figure(), fig.add_axes() и ax1 = fig.add_subplot(111)?

Большое спасибо!:)

Answer 1

Вы должны удалить последние две строки из вашей функции. Эти строки необходимо вызывать после определения всех фигур и подзаговоров.

plt.show()
plt.close()

Например, после удаления этих строк из функции вы можете вызывать функцию с использованием разных вспомогательных участков:

plt.subplot(1,3,1)
bar_plot(df1)
plt.subplot(1,3,2)
bar_plot(df2)
plt.subplot(1,3,3)
bar_plot(df3)

И наконец:

plt.show()
plt.close()

Полагаю, это может сработать.

Answer 2

Предлагаемый ответ действительно решает проблему для показанного графика в консоли.Однако, для файла, сохраненного с помощью plt.savefig(filename.png), команда bbox_inches='tight' делает свое дело удобно:

plt.savefig(filename.png, bbox_inches='tight')