Python Matplotlib根据斜率将散点图分为几类

Question

我有一个x列表和一个y值列表。 我想在Matplotlib中构造一个散点图，然后根据点的x和y坐标将点分为五个类别，如下图所示：

angles = [0, 18, 36, 54, 72, 90]
colors = ['r','g','b','c']
x = [....]
y = [....]

划分类别中的所有点将具有相同的颜色。 对于类别有一个图例也将很棒。 我是Matplotlib和Python的新手，有人知道我该怎么做吗？

Answer 1

这是一个有效的示例，它将为您提供一些入门知识：

from matplotlib import pyplot as plt
from matplotlib.lines import Line2D

import math
import random

fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)

w, h = 7, 5
a = [[random.random() * w, random.random() * h] for i in range(100)]
plt.plot(*zip(*a), marker='o', color='r', ls='')

for deg in [18, 36, 54]:
    r = 10
    line = Line2D([0, r * math.cos(math.radians(deg))],
                  [0, r * math.sin(math.radians(deg))],
                  linewidth=1, linestyle="-", color="green")
    ax.add_line(line)

ax.set_xlim(0, w)
ax.set_ylim(0, h)
plt.legend()
plt.show()