顏色的Python（seaborn）：不增加數據幀

Question

slov = {'People': {0: 'Ivan', 1: 'John', 2: 'Peter', 3: 'Ming'}, 'Country':{0: 'Russia', 1: 'USA', 2: 'USA', 3: 'China'},\ 
    'Height': {0: 181, 1: 175, 2: 174, 3: 173}} 

顏色我想看到這幅畫

但我不知道該怎麼做。我的意思是我想爲俄羅斯人提供紅色，美國人爲綠色，中國人爲黃色。

我學嘗試找到解決：

import numpy as np 
import seaborn as sns 
import matplotlib.pyplot as plt 
import pandas as pd 
sns.set(style="white") 

slov = {'People': {0: 'Ivan', 1: 'John', 2: 'Peter', 3: 'Ming'}, 'Country':{0: 'Russia', 1: 'USA', 2: 'USA', 3: 'China'},\ 
'Height': {0: 181, 1: 175, 2: 174, 3: 173}} 

obj = pd.DataFrame(slov) 

palette=["g", "b", "r"] 
obj['Color']='r' 

row_index = obj.Country == 'Russia' 
obj.loc[row_index, 'Color'] = 'r' 

row_index = obj.Country == 'USA' 
obj.loc[row_index, 'Color'] = 'g' 

row_index = obj.Country == 'China' 
obj.loc[row_index, 'Color'] = 'y' 


g = sns.factorplot(x="People", y="Height", data=obj, kind='bar', palette=obj['Color']) 

plt.show() 

也許我的解決方案是不是很好。我給DataFrame添加了顏色。也許我們可以寫得更好。也許我不需要爲我的DataFrame添加顏色（看起來不是很正確。）。但是，如何在不向DataFrame中添加這些顏色的情況下解決我的任務？

Answer 1

您可以通過dict使用map：

d = {'Russia':'r', 'USA':'g','China':'y'} 
g = sns.factorplot(x="People", 
        y="Height", 
        data=obj, 
        kind='bar', 
        palette=obj['Country'].map(d)) 

plt.show()