python - 在 Python 中为散点图设置图例

Question

我已经将一些数据从 excel 文件导入到 Python 中，我想做一个散点图。我遇到的问题是传说。我包括下面的代码:

data_2 = xlrd.open_workbook('/Users/.../.../Book1.xlsx')
sheet = data_2.sheet_by_index(0)
sheet.cell_value(0,0)
fig = plt.figure(figsize = (11,8), facecolor='white', edgecolor='black')
plt.xlabel('some x label')
plt.ylabel('some y label')
plt.xticks(np.arange(0.0, 30, 1.0))
plt.yticks(np.arange(-9.50, 1.0, 0.50))
plt.axhline(y=0.0, color='grey', linestyle=':')
plt.margins(0.01)
plt.savefig('plot.png')

for i in range(1, sheet.nrows):
    xval = sheet.cell_value(i,0)
    for j in range(1, sheet.ncols):
        yval = sheet.cell_value(i,j)
        plt.scatter(xval,yval, label = 'i = {}'.format(j))
plt.legend(loc='center left', bbox_to_anchor=(1, 0.5)) #the legend will appear on the right of the figure

虽然散点图很好，但图例条目重复了 31 次(31 是数据集中的列数)。我无法弄清楚是什么导致了问题。我尝试了 plt.legend(..) 行的不同位置，但没有成功。我曾尝试从头开始构建图例，但很难从图中提取点的颜色!如何修改代码以仅在右侧使用 sheet.ncols - 1(数据集中的第一列是 xaxis 值)条目输出图例？
如果有帮助，这是我使用 Excel 绘制数据时得到的图表(但我不想使用 Excel 绘制图表)

我希望代码在 python 中重现这样的情节。请注意，图例位置在底部，但最好将其放在图的右侧

Answer 1

尝试只标记一行的条目，例如 plt.scatter(xval,yval, label = 'i = {}'.format(j) if i==1 else None) 。无论标签的内容如何，​​Matplotlib 都会为图中的每个标记对象自动创建一个图例条目。