Можно ли исключить столбец массива numpy из включения в тепловую карту? - PullRequest
Новая
       23

Можно ли исключить столбец массива numpy из включения в тепловую карту?

0 голосов
/ 31 марта 2020

У меня есть массив NumPy (отображающий когортный анализ), и я хочу создать из него тепловую карту. первый столбец показывает итоги каждой когорты, и я хочу, чтобы они отображались не зеленым, а белым цветом.

Мой код выглядит следующим образом: 

colors=[[0.0, 'rgb(160,191,102)'], [.1, 'rgb(180,212,145)'],[.2, 'rgb(220,232,184)'],[.3, 'rgb(246,227,181)'],
            [.4, 'rgb(254,221,139)'], [.5, 'rgb(250,177,23)'],[.6, 'rgb(234,142,120)'],[.7, 'rgb(241,102,81)'],
            [.8, 'rgb(175,101,80)'],[.9, 'rgb(175,101,80)'],[1.0, 'rgb(175,101,80)']]

z_text = np.around(normalized_cohort_array, decimals=2)
z_text = np.insert(z_text, 0, cohort_sizes.astype(np.float64), axis=1)
x.insert(0, 'Total')
fig = ff.create_annotated_heatmap(z_text[::-1,:],x=x, y=y[::-1],  colorscale=colors, connectgaps = False,  zmax = 1.0, zmin = 0.0, reversescale=True)


fig.show()
...