如何在热图中沿对角线值为熊猫数据框着色

时间:2020-09-16 16:31:56

标签: python pandas seaborn heatmap

我有这个excel(Excel File),它由92行组成。我想要一个热图,该热图将沿对角线(从第0行开始,在第61行结束)为excel上色。每个对角线都包含正值和负值。我的要求是将值的颜色设置为'0'=白色,然后正值将显示为红色值越大,红色显示越深),和负值将显示为蓝色值越小,蓝色显示越深)。最后一个对角线从第62行开始。因此,热图将呈平行四边形形状。我可以使用以下代码捕捉数据框的对角线,但是我不确定如何根据上述要求为它们着色。我正在共享用于捕获对角线值的代码,感谢您对它们进行着色的任何帮助。

import pandas as pd
import numpy as np
cols = pd.read_csv("demo_Data.csv",sep=',',header=None).columns
data = pd.read_csv("demo_Data.csv",sep=',',usecols=cols[1:],header=None)

data_np = data.to_numpy()

data_list = data_np.tolist()

dig_list =[]
#catching the diagonals 
for j in range(0,len(data_list)-30):
  temp_list =[]
  for i in range(0,31):
    #temp_list contains every diagonals.
    temp_list.append(data_list[j+i][i])

0 个答案:

没有答案