python - 如何在一个参数中为python中的散点函数设置颜色和标记

Question

我尝试根据标签对一系列数据使用不同的颜色和标记：

我写道：

   for x,l in zip(X,labels):
        ax.scatter(x[0],x[1], 'xb' if l == -1 else 'or')

“xb”是指标记的“x”和颜色的“b”，但它不起作用。

所以我写成：

for x,l in zip(X,labels):
    ax.scatter(x[0],x[1], marker='x' if l == -1 else 'o', c='r' if l == -1 else 'b')

不使用两个单独的命名参数的正确参数是什么？或任何其他缩短语句的技巧。

Answer 1

到目前为止，我还没有在绘图函数中看到 if-else 构造。我会更改标签列表：

X = [(1, 4), (2, 6), (3, 2), (4, 8), (5, 3), (6,1)]
labels = [-1, 0, 0, -1, 0, -1]

fig, ax = plt.subplots()
labels = ["xb" if item == -1 else "or" for item in labels]

for x,l in zip(X, labels):
    ax.plot(x[0],x[1], l)

plt.show()

还值得注意的是，scatter 和 plot 对标记定义使用不同的约定。对于 scatter，plot 的快捷方式不起作用，其中 size s、color c 和 markerstyle m 必须单独提供。