Python - Gráfico de coeficiente de correlación

Python dibuja un mapa de calor del coeficiente de correlación. 

import matplotlib.pyplot as plt
import seaborn as sns

train_corr = data_train1.corr() #计算相关系数

ax = plt.subplots(figsize=(20, 16)) #调整画布大小
cmap = sns.diverging_palette(180, 30, as_cmap=True)  #颜色
ax = sns.heatmap(train_corr, vmax=.8, cmap=cmap,square=True, annot=True)#画热力图 ，annot=True 显示系数