matplotlib:多维datafram的二维散点图

Question

我有下面的数据框架，显示了多个变量，包括年份范围、性别、国籍和人口总数。你知道吗

+---------+---+---+----------+
| range   |Sex|NOC|sum(count)|
+---------+---+---+----------+
|2000-2020|  M|SUR|        13|   
|2000-2020|  M|SRI|        33|   
|1940-1960|  F|YUG|        86|    
|1980-2000|  F|IND|        54|    
|1900-1920|  M|GBR|      1636|    
|2000-2020|  M|AFG|        13| 
+---------+---+---+----------+

我想在matplotlib或seaborn中绘制以下散点图。我该怎么做

Answer 1

您需要取消堆叠sex列以获得另外两列：M和F。肮脏的例子：

>>> df_ = df.set_index(['year', 'noc', 'sex']).unstack().reset_index()
>>> df_.columns = [''.join(col).strip() for col in df_.columns.values]

然后您可以使用seaborn：

>>> sb.scatterplot(x='sumM', y='sumF', hue='year', data=df)