я есть некоторые базовые данные о размере двигателя, лошадиных силах и типе кузова (образец показан ниже)

         body-style  engine-size  horsepower
0   convertible          130       111.0
2     hatchback          152       154.0
3         sedan          109       102.0
7         wagon          136       110.0
69      hardtop          183       123.0

Из этого я сделал точечный график с мощностью в лошадиных силах по оси x, размером двигателя по оси y и использованием стиля кузова в качестве цветовой схемы для дифференциации классов тела и. Я также использовал «степень сжатия» каждого автомобиля из отдельного кадра данных, чтобы определить размер точки.

Это сработало хорошо, за исключением того, что я не могу отобразить цветные легенды для своего сюжета. Помощь нужна, так как я новичок.

Вот мой код:

dict = {'convertible':'red' ,  'hatchback':'blue' , 'sedan':'purple' , 'wagon':'yellow' , 'hardtop':'green'}

wtf["colour column"] = wtf["body-style"].map(dict)
wtf["comp_ratio_size"] = df['compression-ratio'].apply ( lambda x : x*x)

fig = plt.figure(figsize=(8,8),dpi=75)
ax = fig.gca()
plt.scatter(wtf['engine-size'],wtf['horsepower'],c=wtf["colour column"],s=wtf['comp_ratio_size'],alpha=0.4)
ax.set_xlabel('horsepower')
ax.set_ylabel("engine-size")
ax.legend()