Jupyter Notebook: pyplot в выходной ячейке

Question

Я бы хотел, чтобы мои графики появлялись в выходной ячейке, а не в входной ячейке в Jupyter Notebook.Есть ли способ сделать это простым способом?Я смог сделать это, только обернув функцию вокруг графика, но я думаю, что есть лучший способ.

import matplotlib.pyplot as plt
import pandas as pd

iris = pd.read_html('https://en.wikipedia.org/wiki/Iris_flower_data_set')[0].iloc[:, 1:]

fig, ax = plt.subplots(figsize=(12, 7))
for species, data in iris.groupby('Species'):
    ax.scatter(data['Sepal length'], data['Sepal width'],
               label=species, s=100, alpha=0.7)
ax.set_title('Iris-dataset', fontsize=16)
ax.set_xlabel('Sepal Length')
ax.set_ylabel('Sepal Width')
plt.legend()
plt.show()

def my_fig():
    fig, ax = plt.subplots(figsize=(12, 7))
    for species, data in iris.groupby('Species'):
        ax.scatter(data['Sepal length'], data['Sepal width'],
                   label=species, s=100, alpha=0.7)
    ax.set_title('Iris-dataset', fontsize=16)
    ax.set_xlabel('Sepal Length')
    ax.set_ylabel('Sepal Width')
    plt.legend()
    plt.close()
    return fig

my_fig()