以下に、pandasを使用してヒートマップを作成する方法を示します。
まず、pandasと必要な他のライブラリをインポートします。
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
次に、データを読み込みます。例として、CSVファイルからデータを読み込む方法を示します。
data = pd.read_csv('data.csv')
データを適切に整形し、必要なカラムや行を選択します。
selected_data = data[['column1', 'column2', 'column3']]
ヒートマップを作成するために、データの相関行列を計算します。
correlation_matrix = selected_data.corr()
最後に、seabornライブラリを使用してヒートマップをプロットします。
sns.heatmap(correlation_matrix, annot=True, cmap='coolwarm')
plt.title('Correlation Heatmap')
plt.show()
これにより、データの各列(カラム)間の相関関係が視覚化されます。ヒートマップには、相関係数の値が色で表され、相関関係の強さや方向を示します。注釈(annot)を使用すると、各セルに相関係数の値が表示されます。
以上が、pandasを使用してヒートマップを作成する基本的な手順です。実際のデータに応じて、さまざまなカスタマイズオプションを使用することもできます。