熱力圖，又名相關(guān)系數(shù)圖。根據(jù)熱力圖中不同方塊顏色對應(yīng)的相關(guān)系數(shù)的大小，可以判斷出變量之間相關(guān)性的大小。

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

# 樣例數(shù)據(jù)
np.random.seed(0)
n_rows=32
data={
'mpg':np.random.uniform(10,35,n_rows),
'cyl':np.random.choice([4,6,8],n_rows),
'disp':np.random.uniform(100,500,n_rows),
'hp':np.random.uniform(50,350,n_rows),
'drat':np.random.uniform(2,6,n_rows),
'wt':np.random.uniform(1.5,5.5,n_rows),
'qsec':np.random.uniform(14,25,n_rows),
'vs':np.random.choice([0,1],n_rows),
'am':np.random.choice([0,1],n_rows),
'gear':np.random.choice([3,4,5],n_rows),
'carb':np.random.choice([1,2,3,4,6,8],n_rows)
}
df=pd.DataFrame(data)

# 繪圖
plt.figure(figsize=(12,10),dpi=80)
sns.heatmap(df.corr(),xticklabels=df.corr().columns,yticklabels=df.corr().columns,cmap='RdYlGn',center=0,annot=True)
plt.title('Correlogram of mtcars',fontsize=22)
plt.xticks(fontsize=12)
plt.yticks(fontsize=12)
plt.show()