Тепловая карта с несколькими цветовыми картами по столбцам ⇐ Python

[Anonymous]

У меня есть фрейм данных, в котором каждый столбец содержит значения, которые считаются «нормальными», если они попадают в интервал, который различен для каждого столбца:

Код: Выделить всё

# The main df
df = pd.DataFrame({"A": [20, 10, 7, 39],
"B": [1, 8, 12, 9],
"C": [780, 800, 1200, 250]})

df_info представляет интервалы для каждого столбца df.
Так, например, df_info["A"][0] — это min для столбца df["A"], а df_info["A"][1] представляет собой max для столбца df["A"] и так далее.

Код: Выделить всё

df_info =  pd.DataFrame({"A": [22, 35],
"B": [5, 10],
"C": [850, 900]})

Благодаря этому SO-ответу я смог создать собственную тепловую карту для печати синими значениями ниже диапазона, красными значениями выше диапазона и белыми значениями внутри диапазона. Просто помните, что каждый столбец имеет свой диапазон. ТАК я нормализовал в соответствии с этим:

Код: Выделить всё

df_norm = pd.DataFrame()
for col in df:
col_min = df_info[col][0]
col_max = df_info[col][1]
df_norm[col] = (df[col] - col_min) / (col_max - col_min)

И наконец распечатал свою тепловую карту

Код: Выделить всё

vmin = df_norm.min().min()
vmax = df_norm.max().max()

norm_zero = (0 - vmin) / (vmax - vmin)
norm_one = (1 - vmin) / (vmax - vmin)
colors = [[0, 'darkblue'],
[norm_zero, 'white'],
[norm_one, 'white'],
[1, 'darkred']
]
cmap = LinearSegmentedColormap.from_list('', colors, )
fig, ax = plt.subplots()

ax=sns.heatmap(data=data,
annot=True,
annot_kws={'size': 'large'},
mask=None,
cmap=cmap,
vmin=vmin,
vmax=vmax) \
.set_facecolor('white')

В примере вы можете видеть, что третий столбец имеет значения намного выше/ниже по сравнению с интервалом 0-1 (и с первым столбцом), поэтому они «поглощают» все оттенки красного и синего. .
ВОПРОС:
Я хочу добиться использования всех оттенков красного/синего для каждого столбца или, по крайней мере, уменьшить разницу восприятия. между (например) первым и третьим столбцом.
Мне пришлось нелегко:

• создать собственную цветовую карту, где каждая цветовая карта нормализация выполняется по столбцу
• использует несколько цветовых карт, каждая из которых применяется к отдельному столбцу
• применяет цветовую карту mpl.colors.LogNorm, но я не уверен, как использовать его с моей собственной LinearSegmentedColormap

Подробнее здесь: https://stackoverflow.com/questions/668 ... -by-column