pythonの 関数を使ったグラフ出力

Question

大学一年生です。課題が出たので
下記のようなコードを用いて様々なグラフを作成したいです。

import matplotlib.pyplot as plt
plt.figure(ifgsize=(20,10))

#ここに様々なグラフ処理

plt.xlabel.('x label')
plt.ylabel('y label')
plt.grid(True)

上のような処理を繰り返すのも綺麗ではないと思うので、これを関数化したらいいのかと考えました。

import matplotlib.pyplot as plt


def plot_model(plot_method):
  plt.figure(figsize=(20,10))
  
  plot_method

  plt.xlabel('x label')
  plt.ylabel('y label')
  plt.grid(True)

plot_model(plt.hist(math_data['absences']))
plot_model(plt.plot(math_data['absences']))

そうすると、確かにフォーマットで、グラフは出力されるのですが、
下図のように、いちばん上のグラフはフォーマットを適応されずに、なぜか三番目のグラフが出力されました。どのようにしたら２つのグラフを適切なフォーマットで出力されますか？

Answer 1

現状の問題点は、TakaiYさんの指摘の通りなので、
方法としては、プロットの関数と、その引数を別々に渡して、関数の中でプロットの関数を実行するようにしたら良いです。

import matplotlib.pyplot as plt

def plot_model(plot_method, data):
  plt.figure(figsize=(20, 10))

  plot_method(data)

  plt.xlabel('x label')
  plt.ylabel('y label')
  plt.grid(True)

plot_model(plt.hist, math_data['absences'])
plot_model(plt.plot, math_data['absences'])

Answer 2

詳細にどのようなことが起きているか追っているわけではありませんが、

plot_model(plt.hist(math_data['absences']))

ここで、指定しているplt.hist(math_data['absences'] の部分は、関数内部の「plot_method」のところでなくて、 この 呼び出しの時点で実行されてしまっています。

そして、plot_model 関数の引数に渡されているのはplt.hist関数の返り値です。
想定通りplot_model関数の中で引数で与えている処理をさせたいということであれば、もうちょっと工夫する必要があります。