matplotlib меняет цвет набора данных

Question

import pandas as pd
import matplotlib.pyplot as plt

iris = pd.read_csv(r'C:\Users\MACHINE\Desktop\iris.csv', names=['sepal_length', 'sepal_width',         
'petal_length', 'petal_width', 'class'])

colors = {'Versicolor':'r', 'Setosa':'g', 'Virginica':'b'}
# create a figure and axis
fig, ax = plt.subplots()
# plot each data-point
for i in range(len(iris['sepal_length'])):
    ax.scatter(iris['sepal_length'][i], iris['sepal_width'][i], color= colors[iris['class'][i]])
# set a title and labels
ax.set_title('Iris Dataset')
ax.set_xlabel('sepal_length')
ax.set_ylabel('sepal_width')


ax.scatter(iris['sepal_length'][i], iris['sepal_width'][i], color= colors[iris['class'][i]])

KeyError: 'class' Я новичок в python, скорее всего, это синтаксическая ошибка. Я также дал ссылку на сайт, на котором копировал код с https://towardsdatascience.com/introduction-to-data-visualization-in-python-89a54c97fbed

Answer 1

for i in range(len(iris['sepal_length'])):
    ax.scatter(iris['sepal_length'][i], 
    iris['sepal_width'[i],color=colors.get(iris['class'][i]) )

используйте метод get вместо