Можете ли вы поместить условный statemnet внутри команды plot в Python?

Question

У меня есть набор данных (два столбца «еда 1», «еда 2» + метка столбец «вкус»).

Я хочу построить два столбца как X и Y и в зависимости от меткиЯ хочу изменить форму маркера. Столбец метки является двоичным [0, 1].

для примера:

data.plot(kind='scatter', x='food 1', y='food 2')

например, если метка равна '0', я хочу использовать квадратный маркер, а если '1'Я хочу использовать маркер треугольника.

Как это сделать?

Answer 1

Не существует API для применения условных обозначений к маркерам в одном вызове scatter.

Обычное решение - разделить точки на две группы и дважды вызвать scatter () ,один раз для 0 отмеченных точек и еще раз для 1 отмеченных точек:

plt.scatter(x0, y0, marker='s')
plt.scatter(x1, y1, marker='^')

Answer 2

Вы можете делать встроенные операторы if-else в Python, например:

marker='^' if (label=='0') else 's'