Colorbar for barplot, когда цвета бара выбираются по рангу

Question

Я пытаюсь создать барплот, где цвета баров будут выбраны в соответствии с заданными значениями из предопределенного набора цветов. Значение «pahse» определяет цвет. Он разделен на 3 диапазона и диапазоны ранжируются. Ранг используется для вызова палитры colorfrom. Эта часть работает нормально. Но я застрял в цветовой гамме. Цветная панель должна содержать цвета на основе клавиш словаря «rank_classes» и tiks/labels на основе значений, которые использовались для ранжирования. Я попробовал mpl.colorbar.plorbarBase(), но не похоже на seaborns sns.cubehelix_palette(). Интересно, как можно объединить эти две вещи или предложить более простые решения?

import pandas as pd 
import numpy as np 
import matplotlib.pyplot as plt 
import seaborn as sns 
sns.set_style("white") 
#%matplotlib inline 

index = ['25', '26', '27'] 
count = [10, 50, 22] 
phase = [0.9, 2.2, 1.2] 
ranks = [0, 2, 1] 

rank_classes = {0:"Ph<1", 1:"1<Ph<1.2", 2:"Ph>=1.2"} 

d = {'count' : pd.Series(count, index=index), 
    'phase' : pd.Series(phase, index=index), 
    'rank' : pd.Series(ranks, index=index) 
    } 

df = pd.DataFrame(d) 

# barplot 
fig, ax = plt.subplots() 
pal = sns.cubehelix_palette(3) 

x = df.index 
y = list(df['count']) 
z = list(df['rank']) 

sns.barplot(x=x, y=y, palette=np.array(pal)[z]) 
sns.despine() 

Answer 1

Почти там

import numpy as np 
import matplotlib as mpl 
import matplotlib.pyplot as plt 
import seaborn as sns 
sns.set_style("white") 
%matplotlib inline 

index = ['25', '26', '27'] 
count = [10, 50, 22] 
phase = [0.9, 2.2, 1.2] 
rank = [0, 2, 1] 
# 
rank_classes = {0:"Ph<1", 1:"1<Ph<1.2", 2:"Ph>=1.2"} 

pal = sns.cubehelix_palette(len(index)) 
cmp = mpl.colors.LinearSegmentedColormap.from_list('my_list', pal,\ 
                N=len(index)) 
plot = plt.scatter(index, count, c=range(len(index)), cmap=cmp) 
plt.clf() 
plt.colorbar(plot) 
sns.barplot(x=index, y=count, palette=np.array(pal)[rank]) 
sns.despine() 

Производит хороший участок с Colorbar. Единственное, что мне нужно выяснить, - это заменить метки ярлыков цветной панели (0, 0.3 ... 3.0) на строки, перечисленные в словаре rank_classes.