区分Python图中的某些点

Question

我在python中有一个情节，

a = (1,2,3,4,5)
b = (1,2,3,4,5)
plot(a,b)

我想用独特的颜色点来区分图中的某些x轴点 例如，

c = (2,4)

我尝试了以下几点：

a = (1,2,3,4,5)
b = (1,2,3,4,5)
plot(a,b)
matplotlib.pyplot.scatter(a, c)

但是我遇到了错误“ x和y必须是相同的大小”

Answer 1

如果您Google Google“ matplotlib scatter”或类似的东西，则第一个链接之一将是这个，它说该功能给出了“ y vs. x的散点图”。因此，您共享的错误消息是有道理的，因为a的长度大于c的长度。我希望对您有意义，为什么您的原始代码会出现该错误。

您的问题已经足够具体，以至于我知道matplotlib中没有开箱即用的解决方案，因此这需要使用您自己的自定义功能。我会提供一种可能对您有帮助的方法。 我在这里构建答案，以便您可以看到如何解决自己的规格，而不是过于依赖复制＆amp;粘贴其他人的代码，因为后者使您更难做到要做您想做的事情。

更简洁地重述问题： matplotlib如何用户绘制一条线，然后将标记放在线的点子集上，由其x值指定？

开始，这是您的程序当前可能的样子：

import matplotlib.pyplot as plt

x_coords = [ ] # fill in x_coords here
y_coords = [ ] # fill in y_coords here
marker_x_coords = [ ] # fill in with x coordinates of points you want to have markers

plt.plot(x_coords, y_coords) # plots the line

#### TODO: plot the markers ####

现在，您拥有该程序的X值您想放上标记的点。您如何获得他们相应的Y值？好吧，您可以制作一个函数，以在x_coords中搜索X值的索引，然后在y_coords的相同索引下给出相应的值：

def getYVals(x_coords_lst, y_coords_lst, marker_x_coords_lst):
    marker_y_coords = [] # start with an empty list
    for x_point in marker_x_coords_lst:
        point_index = x_coords_lst.index(x_point) # get the index of a point in the x list
        marker_y_coords.append(y_coords_lst[point_index]) # add the value of the y list at that index to the list that will be returned
    return marker_y_coords

这是' t最快的方法，但这是最清晰的方法。这是一个替代方案，可以给出相同的结果，但可能会在计算上执行更快的速度（它使用“列表理解”）：

def getYVals(x_coords_lst, y_coords_lst, marker_x_coords_lst):
    return [y_coords_lst[x_coords_lst.index(x_val)] for x_val in marker_x_coords_lst]

以上任何一个getyvals函数的输出都可以作为y值的y值标记。可以将其放在y plt.scatter的参数中，并且您已经具有x值，因此您应该从那里开始好！