如何在散点图中将异常值更改为其他一些颜色

Question

如果我有这样的散点图

我想知道有什么方法可以将明显的异常值（例如顶部的三个）更改为同一图中的其他颜色？

Answer 1

首先，您需要找到“异常值”的标准。 一旦你有了它，你就可以在你的情节中掩盖那些不需要的点。 在 numpy 中可以轻松地根据条件选择数组的子集，例如，如果a是 numpy 数组，则a[a <= 1]将返回所有值大于 1 的数组“切出”。

然后可以按如下方式进行绘图

import numpy as np
import matplotlib.pyplot as plt

num= 1000
x= np.linspace(0,100, num=num)
y= np.random.normal(size=num)

fig=plt.figure()
ax=fig.add_subplot(111)
# plot points inside distribution's width
ax.scatter(x[np.abs(y)<1], y[np.abs(y)<1], marker="s", color="#2e91be")
# plot points outside distribution's width
ax.scatter(x[np.abs(y)>=1], y[np.abs(y)>=1], marker="d", color="#d46f9f")
plt.show()

生产

在这里，我们从正态分布绘制点，对分布宽度之外的所有点进行不同的着色。

Answer 2

ImportanceOfBeingErnest有一个很好的答案。 如果我有一个与数据点的枚举类别相对应的数组（在可视化预先划分为类的数据时特别有用），那么这是我使用的单行。

import numpy as np
import matplotlib.pyplot as plt

num = 1000
x= np.random.rand(1,100)
y= np.random.rand(1,100)*2

# Creating a simple data point classification criteria, classes in this case will be 0, 1 and 2
classes = np.round(y)

# Passing in the classes for the "c" argument is super convinient
plt.scatter(x,y, c=classes,cmap=plt.cm.Set1)
plt.show()

将图形划分为 3 个彩色区域的相应散点图：