我有一些数据可以说x
,y
和z
。所有都是1D
数组。我用z
作为颜色的散点图;
import matplotlib.pyplot as plt
plt.scatter(x,y,c=z,alpha = 0.2)
plt.xlabel("X")
plt.ylabel("Y")
plt.ylim((1.2,1.5))
plt.colorbar()
z
值已归一化,且在-1
到1
之间。我附上了下图。
我的问题是;如何过滤颜色,以便让颜色值介于-0.25
到0.25
之间的点从图中消失(即将颜色设置为白色)。
如果需要回答此问题,可以提供x
,y
和z
的值。感谢您的时间。
答案 0 :(得分:1)
import matplotlib.pyplot as plt
import numpy as np
np.random.seed(42)
# prepare random data
stats = -1, 1, 200
x = np.random.uniform(*stats)
y = np.random.uniform(*stats)
z = np.random.uniform(*stats)
# mask unwanted data
thresh = 0.4
mask = np.abs(z) <= thresh
x_ma = np.ma.masked_where(mask, x)
y_ma = np.ma.masked_where(mask, y)
z_ma = np.ma.masked_where(mask, z)
并进行绘图:
fig, (ax_left, ax_right) = plt.subplots(1, 2, figsize=(10, 4),
sharex=True, sharey=True)
img_left = ax_left.scatter(x, y, c=z)
fig.colorbar(img_left, ax=ax_left)
img_right = ax_right.scatter(x_ma, y_ma, c=z_ma)
fig.colorbar(img_right, ax=ax_right)
给出以下结果:
右侧的图表隐藏了低于所选阈值的所有点。