以下のコードでデータセットが表示されません

Question

Pythonではじめる機械学習で勉強をはじめました。
github(https://github.com/amueller/introduction_to_ml_with_python/blob/master/01-introduction.ipynb)を元に
以下のコードを書きました

import pandas as pd
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
import mglearn

iris_dataset = load_iris()

X_train, X_test, y_train, y_test = train_test_split(
    iris_dataset['data'], iris_dataset['target'], random_state=0)

iris_dataframe = pd.DataFrame(X_train, columns=iris_dataset.feature_names)

pd.plotting.scatter_matrix(iris_dataframe, c=y_train, figsize=(15, 15),
                           marker='o', hist_kwds={'bins': 20}, s=60,
                           alpha=.8, cmap=mglearn.cm3)

しかし画像が表示されません。
修正点を教えて下さい。

環境
WSL2 Ubuntu 20.04.3 LTS
python 3.8.10

Answer 1

以下のようにplt.show()をしましょう。

import pandas as pd
import matplotlib.pyplot as plt
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
import mglearn

iris_dataset = load_iris()

X_train, X_test, y_train, y_test = train_test_split(
    iris_dataset['data'], iris_dataset['target'], random_state=0)

iris_dataframe = pd.DataFrame(X_train, columns=iris_dataset.feature_names)

pd.plotting.scatter_matrix(iris_dataframe, c=y_train, figsize=(15, 15),
                           marker='o', hist_kwds={'bins': 20}, s=60,
                           alpha=.8, cmap=mglearn.cm3)

plt.show()

Comments

[skru]

回答していただきありがとうございます！
plt.show() をしましたが表示されません。
WSL2上のUbuntuで動かしていることが原因でしょうか？

[jbpb0]

> WSL2上のUbuntu

質問者さん

これの回答を見てください
https://teratail.com/questions/209806