Python 绘制散点密度图_mb61cf0176d3323的技术博客_

1 matplotlib绘制散点密度图

散点密度主要是计算样本点的出现次数，即密度。

import numpy as np
import matplotlib.pyplot as plt
from scipy.stats import gaussian_kde

# Generate fake data
N=1000
x = np.random.normal(size=N)
y = x * 3 + np.random.normal(size=N)

# Calculate the point density
xy = np.vstack([x,y])  #  将两个维度的数据叠加
z = gaussian_kde(xy)(xy)  # 建立概率密度分布，并计算每个样本点的概率密度

# Sort the points by density, so that the densest points are plotted last
idx = z.argsort()
x, y, z = x[idx], y[idx], z[idx]

fig, ax = plt.subplots()
plt.scatter(x, y,c=z, s=20,cmap='Spectral') # c表示标记的颜色
plt.colorbar()
plt.show()

Python 绘制散点密度图_html

scatter中其他可使用的cmap，参考【https://matplotlib.org/tutorials/colors/colormaps.html】

colorbar反向在颜色名称后面加 _r ，比如： cmap='Spectral_r' 。

2 seaborn绘制散点密度图

import seaborn as sns
sns.kdeplot(x=x, y=y, fill=True, cmap='Spectral', cbar=True)

Python 绘制散点密度图_开发语言_02

3 mpl-scatter-density包

# !pip install mpl-scatter-density
import mpl_scatter_density

N=100000
x = np.random.normal(size=N)
y = x * 3 + np.random.normal(size=N)

# 绘制二维散点密度图
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1, projection='scatter_density')
density = ax.scatter_density(x, y, cmap='Spectral_r')
ax.set_xlim(-3, 3)
ax.set_ylim(-10, 10)
fig.colorbar(density, label='Number of points per pixel')
fig.savefig('gaussian.png')

Python 绘制散点密度图_2d_03

使用该包绘制的图零值有颜色，可通过如下方法使零值变为白色：

import matplotlib.colors as mcolors
norm = mcolors.TwoSlopeNorm(vmin=-1, vmax =60, vcenter=0)
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1, projection='scatter_density')
density = ax.scatter_density(x, y,norm=norm, cmap=plt.cm.RdBu)
ax.set_xlim(-3, 3)
ax.set_ylim(-10, 10)
fig.colorbar(density, label='Number of points per pixel')
fig.savefig('gaussian_color_coded.png')

Python 绘制散点密度图_html_04

虽然上述方式能使零值变为白色，但不美观，可将零值掩码或赋值为 nan 。在源码 base_image_artist.py 中的 make_image 函数( 180 行)中加上如下语句：

array = np.where(array>0, array, np.nan)
# 或者
array = np.ma.masked_array(array, mask=(array<=0))

Python 绘制散点密度图_2d_05

该包调用了 fast_histogram 计算样本频率：

from fast_histogram import histogram2d

ymin, ymax = y.min(), y.max()
xmin, xmax = x.min(), x.max()

array = histogram2d(y, x, bins=10,range=((ymin, ymax), (xmin, xmax)))

推荐文章

眉毛粗的毛衣 · AttributeError: module 'matplotlib' has no attribute 'rcParams_Fix for matplotlib rcParams error_ -

1 月前

飞奔的松树 · matplotlib.mlab.bivariate_normalの使い方 - シリコンの谷のゾンビ

1 月前

瘦瘦的棒棒糖 · OpenCV 4基础篇| OpenCV图像基本操作开发者社区

2 周前

满身肌肉的拖把 · “30万投《水门桥》可赚百万”“易烊千玺主演差不了”…小心电影投资也有杀猪盘|易烊千玺_新浪财经_新浪网

6 月前

很酷的蚂蚁 · PAT(Advanced Level)刷题指南 —— 第八弹_天使Di María的博客-CSDN博客

2 年前

求醉的斑马 · 【FGO】官方同人漫画《魔女的某一天》汉化（喀耳刻玉藻猫） - 哔哩哔哩

2 年前

憨厚的水煮鱼 · 贾樟柯监制影片《不止不休》曝演员阵容白客苗苗演绎真实北漂故事_新闻频道_央视网(cctv.com)

2 年前

儒雅的甘蔗 · 台大学生虐杀两猫引发公愤被合并判刑10个月|台大学生|虐猫_新浪新闻

2 年前