Pythonでデータをヒートマップで可視化したい

Question

やりたいこと

機械学習を勉強しています。その過程で、2つの変数をまとめた1つの説明変数 (x1, x2) と 1つの目的変数 y の相関を見たいです。

野球で例えるなら、x1 がストライク数、x2 がボールの数、y がピッチャーの投げる球種(数字で表す)、といった感じです。今与えられているのは x1 と x2 で、現在の状況を表す (x1, x2) と y の相関を見たいということです。

試していること

今、説明変数 X=(x1, x2) は2次元配列にしていて、y は 1次元配列です。これをmatplotlibかseabornを用いて x1 と x2 を軸とするヒートマップにして表示しようとしているのですが、うまくいきません。

X = [[1, 2],
     [2, 3],
     [2, 2]]

y = [1, 2, 3]

変数はこのような状態です。この場合、X の各値で指定される場所に y の各値をマッピングしたいです。ヒートマップではなくても、相関を見れる良い方法があれば教えていただきたいです。
よろしくお願いします。

Answer 1

こんな感じ？

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

X = [[1, 2],
     [2, 3],
     [2, 2]]

y = [1, 2, 3]

df = pd.DataFrame(X, columns=['x1', 'x2'])
df['y'] = y

df_pivot = pd.pivot_table(data=df, values='y', columns='x1', index='x2', fill_value = 0)

plt.figure()
sns.heatmap(df_pivot, annot=True, fmt="d")
plt.show()

Comments

[feyn]

ありがとうございます！
pivot_tableなる便利な関数があるんですね！
勉強になりました。