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)
我想把它做成16个不同大小的点。基本上,我想计算两列的总和并确定适当的磅值。
答案 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')