python - 你可以在 Python 的绘图命令中放置一个条件语句吗？

Question

我有一组数据（两列'食物1'，'食物2'+标签列'味道'）。

我想将两列绘制为 X 和 Y，并根据我想更改标记形状的标签。标签列是二进制 [0, 1]。

例如：

data.plot(kind='scatter', x='food 1', y='food 2')

例如，如果标签是“0”，我想使用方形标记，如果是“1”，我想使用三角形标记。

怎么做？

标签： pythonpandasmatplotlib

Answer 1

没有一个 API 可以在一次 scatter 调用中将条件应用于标记。

通常的解决方案是将点分成两组并调用scatter()两次，一次用于标记为 0 的点，另一次用于标记为 1 的点：

plt.scatter(x0, y0, marker='s')
plt.scatter(x1, y1, marker='^')