ヒートマップの表示(python)

Question

実現したいこと

https://mori-memo.hateblo.jp/entry/2022/02/23/003111
と同様のことがやりたいです。

発生している問題・エラーメッセージ

上記サイトと同じコードで実行しましたが、うまくいきません。コードの実行はできるのですが、左のヒートマップを押しても散布図が表示されません。

該当のソースコード

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
from sklearn.datasets import load_boston


cols = dd.columns
rows = cols
corr = dd.corr()

def onclick(event):
    x = event.xdata
    y = event.ydata
    axes = event.inaxes
    # axis外クリック時は無視
    if x == None or y == None:
        return
    # ヒートマップ以外をクリック時は無視
    if axes != ax[0]:
        return

    # x、ｙの小数点切り捨ての値がインデックス番号と一致する
    col = cols[int(x)]
    row = rows[int(y)]

    # 散布図の範囲指定用に描くデータの最大値-最小値を計算しておく
    xwid = dd[col].max() - dd[col].min()
    ywid = dd[row].max() - dd[row].min()

    # 右の散布図の更新
    ax[1].set_title('Correlation coefficient ' + col + ' - ' + row + ': ' \
                    + str(round(corr.loc[row, col], 3)))
    ax[1].set_xlabel(col)
    ax[1].set_ylabel(row)
    ln.set_data(dd[col], dd[row])
    # 散布図の範囲はmax,min値より少し大きめにする
    ax[1].set_xlim(dd[col].min() - xwid * 0.1, dd[col].max() + xwid * 0.1)
    ax[1].set_ylim(dd[row].min() - ywid * 0.1, dd[row].max() + ywid * 0.1)
    plt.draw()

fig, ax = plt.subplots(1, 2, figsize=(24, 15))
sns.heatmap(corr, ax=ax[0], vmin=-1, vmax=1, annot=True, fmt="3.2f", annot_kws={"size": 5})
ln, = ax[1].plot([np.nan], [np.nan], 'o')  # 散布図プロット
ax[0].set_aspect('equal')
fig.canvas.mpl_connect('button_press_event', onclick)
plt.show()

#実行結果

Comments

[can110]

提示コードはNameErrorで実行できませんでした。
現象が再現するコードを提示ください。
ちなみに参考サイトのコードはこちらでは正常に動作しました。

[ke0313]

サイトのコードと同じものを実行しました。図は出てくるのですが、ヒートマップをクリックしても散布図が表示されない状況です。

[bsdfan]

実行環境は何でしょうか？ jupyter だったりしますか?

[ke0313]

anacondaからJupyterを使ってます。python3.9です。

[jbpb0]

> anacondaからJupyterを使ってます。

もし

%matplotlib inline

を実行してたら、あるいは「%matplotlib ...」をどれも実行してなければ、下記のどちらかをコードの先頭で実行してみてください

%matplotlib notebook
%matplotlib tk

[ke0313]

Vscodeで実行してますが、%matplotlib notebookを先頭にいれました。ただ、同様に動きませんでした。

[jbpb0]

%matplotlib tk

ではどうですか？
別のウインドウが開いて、そこにグラフが表示されるはずですが
(vs codeは使ってないのでよく知りませんが、直接jupyterを使った場合はそうなります)

[ke0313]

%matplotlib tkでは、黄色の下線が引かれて、実行後Warning: Cannot change to a different GUI toolkit: tk. Using notebook instead.　と表示します

Answer 1

jupyter環境で%matplotlib notebookを先頭にいれました