- 积分
- 63538
- 贡献
-
- 精华
- 在线时间
- 小时
- 注册时间
- 2011-7-23
- 最后登录
- 1970-1-1
|
发表于 2022-5-14 22:09:09
|
显示全部楼层
def density_calc(x, y, radius):
"""
散点密度计算(以便给散点图中的散点密度进行颜色渲染)
:param x:
:param y:
:param radius:
:return: 数据密度
"""
res = np.empty(len(x), dtype=np.float32)
for i in range(len(x)):
print(i)
res[i] = np.sum((x > (x[i] - radius)) & (x < (x[i] + radius))
& (y > (y[i] - radius)) & (y < (y[i] + radius)))
return res
radius = 3
Z1 = density_calc(observation, prediction, radius)
plt.scatter(observation, prediction, c=Z1, cmap=colormap, marker=".", s=marker_size.norm=colors.LogNorm(vmin=Z1.min(), vmax=0.5 * Z1.max())) |
|