У меня есть следующий код, который строит точечную диаграмму, которая основывается на категориальной переменной, ее значение определено в цветах.
fig, ax = plt.subplots()
colors = {'SpmcArrayQueue':'red', 'SpscArrayQueue':'blue', 'MpscArrayQueue':'green', 'MpmcArrayQueue':'black'}
ax.scatter(df_qtype['qCapacity'], df_qtype['Measurement'], c=df_qtype['qType'].apply(lambda x: colors[x]))
plt.title(bench)
plt.ylabel('Measurement')
plt.xlabel('qCapacity')
ax.legend(df_qtype['qType'])
plt.show()
Я пытался добавить легенду о категориальной переменной, чтобы понять соответствующие цвета на графике, но у меня есть следующая проблема
