Как я могу исключить колонку из тепловой карты? ⇐ Python

[Anonymous]

У меня есть поворотный флот данных, и я хочу построить тепловую карту в слое. У меня есть общий столбец, что затрудняет использование цветовой шкалы. < /P>

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

import numpy as np
import pandas as pd
import plotly.express as px

city_origin = ['London', 'Tokio', 'Seoul', 'Paris', 'Tashkent', 'Washington', 'Moscow']
city_current = ['London', 'Madrid', 'Tashkent', 'Seoul', 'Paris', 'Toronto', 'Washington', 'Istanbul', 'Hanoi', 'Manilla', 'Delhi', 'Busan', 'Moscow']

migrant_origin = np.random.choice(city_origin,size = 1000)
migrant_current = np.random.choice(city_current, size = 1000)
migrant_salary = np.random.randint(1300, 6900, size = 1000)
df = pd.DataFrame({'migrant_origin':migrant_origin, 'migrant_current':migrant_current, 'migrant_salary':migrant_salary})

new_df = pd.pivot_table(df, index = 'migrant_origin', columns = 'migrant_current', values ='migrant_salary', aggfunc = 'sum', fill_value = 0)

new_df['total'] = new_df.sum(axis = 1)
cols = ['total'] + [col for col in new_df.columns if col != 'total']
new_df = new_df[cols]

px.imshow(new_df,text_auto=True)

Я хочу показать тоталсы, но не затронутые цвета на тепловой карте. Есть ли способ исключить столбцы из окраски?

Подробнее здесь: https://stackoverflow.com/questions/796 ... -a-heatmap