Matplotlib 散点图，每个 x 点有 2 个 y 点

Question

我想使用 Matplotlib 创建散点图。

我有一组用于 x 轴的 ID。它看起来像这样：

x = [1, 2, 3, ...]
#Length = 418

对于 y 轴，我有一个元组数组。它看起来像这样：

y = [(1, 1), (1, 0), (0, 0), (0, 1), ...]
#Length = 418

我想将其绘制在散点图中，以便对于每个 ID（在 x 轴上），它为相应元组的第一个值显示一个红点，为相应元组的第二个值显示一个蓝点.

我试过这段代码：

plt.plot(x, y, 'o')
plt.xlim(xmin=900, xmax=1300)
plt.ylim(ymin=-1, ymax=2)
plt.yticks(np.arange(2), ('Class 0', 'Class 1'))
plt.xlabel('ID')
plt.ylabel('Class')
plt.show()

它显示了这个结果：

如何让这个情节更清晰？

另外，当我放大时，我注意到这些点是偏移的？如何将它们直接放在彼此上方？

Answer 1

我认为你需要整理你的数据，以便拥有平面列表。

我认为最直观的方法是定义两个列表，y1、y2 并循环访问您原来的 y，并将正确的值附加到它们：

y1 = []
y2 = []

for i, j in y:
    y1.append(i)
    y2.append(j)

然后您可以通过两次调用plt.scatter 来绘制红色和蓝色散点图

plt.scatter(x, y1, c = 'r')
plt.scatter(x, y2, c = 'b')

希望有帮助！

Answer 2

您需要将数据存储在两个列表/元组中，而不是一个包含两项元组的列表中，因为这是plt.plot 需要作为输入的内容。最好的方法是像这样重新组织您的数据，但如果您被这个元组列表困住（或者您出于其他原因需要这种数据格式），您可以随时创建列表：

plt.plot(x, [i for (i,j) in y], 'ro')
plt.plot(x, [j for (i,j) in y], 'bo')

请注意，如果您想要两条不同颜色的不同数据线，则需要两条plt.plot(..) 线。

为避免标记相互重叠，您可以更改它们的形状或大小（或两者都更改）：

plt.plot(x, [i for (i,j) in y], 'rs', markersize = 12 )
plt.plot(x, [j for (i,j) in y], 'bo', markeresize = 8)