Условный маркер для диаграммы рассеяния Matplotlib ⇐ Python

[Anonymous]

Я пытаюсь построить график x1 и x2 друг против друга, и если класс == 1, то маркер должен быть +, если класс == 0, то маркер должен быть просто точкой.
Файл.csv структурирован следующим образом: столбец класса, который является условным, будет иметь значение только 1 или 0:



x1
x2
класс




1
2
0


9
4
1


0
5
1


2< /td>
6
0


/div>
Вот код:

Код: Выделить всё

df = pd.read_csv('file.csv')
print(df)
x1 = df['x1'].to_numpy()
x2 = df['x2'].to_numpy()
class = df['class'].to_numpy()

figure, ax = plt.subplots()

for i in range(0,80,1):
if label[i] > 0:
ax.plot(x1[i], color="red", marker='o')
else:
ax.plot(x1[i], color="blue", marker='x')

Вот как ДОЛЖЕН выглядеть результат:



Подробнее здесь: https://stackoverflow.com/questions/790 ... matplotlib