如何在特定位置添加标签以及如何将总和作为图的大小

时间:2019-05-14 14:39:25

标签: python pandas matplotlib data-visualization visualization

pandas DataFrame包含但不限于以下值:

x      y       z    
Winter Night   2222
Spring Morning 3888
Summer Day     1111
Autumn Evening 2999
...

对不起,由于NDA,我无法提供完整的数据集

z变量用于计算总和并使其成为图形上点的大小

def Scatter(x, y, z, colour_data=None):
    f = plt.figure()
    ax.scatter(x, y, s=z)

What I already have

我想把它做成16个不同大小的点。基本上,我想计算两列的总和并确定适当的磅值。

1 个答案:

答案 0 :(得分:0)

@Sheldore很好地说明了您的示例中没有足够的代码可以使用,因此以下内容可能与您所寻找的不完全相同。

我将季节和时间名称转换为整数,因为散点图无法绘制x / y轴的字符串。

import pandas as pd

df = pd.DataFrame(data={
    'x': ['Winter','Spring','Summer','Autumn'],
    'y': ['Night','Morning','Day','Evening'],
    'z': [2222,3888,1111,2999],
})

# convert 'x' and 'y' strings to int to enable scatter plot

season_map = {
    'Winter': 0,
    'Spring': 1,
    'Summer': 2,
    'Autumn': 3,
}

time_map = {
    'Night': 0,
    'Morning': 1,
    'Day': 2,
    'Evening': 3,
}

df.loc[:,'x_int'] = df.x.map(season_map)
df.loc[:,'y_int'] = df.y.map(time_map)

ax = df.groupby(['x_int','y_int']).z.sum().reset_index()\
    .plot.scatter(x='x_int', y='y_int', s=df['z'])

ax.set_xlabel('X')
ax.set_ylabel('Y')

enter image description here