Как построить pd.Series и использовать цвет для каждой точки, поступающей из другого источника данных?

Question

У меня есть pd.Series с именем data, который я хотел бы построить, и у меня есть еще одна серия с таким же индексом, как у data, которая называется colors. Я хотел бы построить data, и для каждой точки графика я хотел бы раскрасить его на основе соответствующего значения в Серии colors (используя некоторую цветовую карту). Можно ли сделать это в matplotlib или seaborn?

Answer 1

Вы можете сделать это с помощью matplotlib, например:

import matplotlib.pyplot as plt

data = pd.Series(range(5))
colors = pd.Series(['r','g','b','b','r'])

plt.scatter(data.index, data, c=colors)
plt.show()

или если цвет серии больше похож на номер группы (не цвета ), вы можете сделать это с помощью seaborn, например:

import seaborn as sns

data = pd.Series(range(5))
colors = pd.Series([0, 1, 2, 2, 0])

sns.scatterplot(data.index, data, hue=colors)