在Python中绘制值组

时间:2016-10-09 07:07:42

标签: python matplotlib plot seaborn python-ggplot

我有一个看起来像这样的数据框。

        country  age  new_user  
298408      UK   32         1   
193010      US   37         0 
164494      UK   17         0   
28149       US   34         0  
297080   China   29         1    

我想在Python中的单个图表中为每个国家/地区绘制new_users的数量(20-30,30-40等)。

基本上,我需要为所有年龄组绘制new_user(值0),为所有国家/地区的所有年龄组绘制new_user(值1)。

我发现很难将年龄分为20-30,30-40等等。 有人可以帮我用python中的seaborn或ggplot或matplotlib来绘制这个吗? ggplot是优选的!

谢谢。

1 个答案:

答案 0 :(得分:1)

import seaborn as sns
from pandas import DataFrame
from matplotlib.pyplot import show, legend
d = {"country": ['UK','US','US','UK','PRC'],
       "age": [32, 37, 17, 34, 29],
       "new_user": [1, 0, 0, 0,1]}

df = DataFrame(d)
bins = range(0, 100, 10)
ax = sns.distplot(df.age[df.new_user==1],
              color='red', kde=False, bins=bins, label='New')
sns.distplot(df.age[df.new_user==0],
         ax=ax,  # Overplots on first plot
         color='blue', kde=False, bins=bins, label='Existing')
legend()
show()

enter image description here