Python皮尔逊相关系数热力图绘制，下三角，上三角

import numpy as np
import seaborn as sns
import pandas as pd
from matplotlib import pyplot as plt
 
# 假设data是一个包含数据的pandas DataFrame
# 这里只是示例，需要根据实际情况调整
data = pd.DataFrame({
    'A': [1, 2, 3, 4, 5],
    'B': [5, 4, 3, 2, 1],
    'C': [2, 3, 4, 5, 6],
    'D': [6, 5, 4, 3, 2]
})
 
# 计算皮尔逊相关系数矩阵
corr_matrix = data.corr()
 
# 绘制皮尔逊热力图
sns.heatmap(corr_matrix, annot=True, square=True, cmap='Blues', mask=np.triu(np.ones_like(corr_matrix)), fmt='.2f')
plt.show()
 
# 下三角和上三角分开处理
# 下三角
mask_lower = np.tril(np.ones_like(corr_matrix))
sns.heatmap(corr_matrix, annot=True, square=True, cmap='Blues', mask=mask_lower, fmt='.2f')
plt.show()
 
# 上三角
mask_upper = np.triu(np.ones_like(corr_matrix))
sns.heatmap(corr_matrix, annot=True, square=True, cmap='Blues', mask=mask_upper, fmt='.2f')
plt.show()

这段代码首先计算皮尔逊相关系数矩阵，然后使用seaborn的heatmap函数绘制皮尔逊热力图，并使用numpy的triu和tril函数来创建掩码，分别显示下三角和上三角。