Легенда точечной диаграммы с цветами для строкового атрибута в сложном фрейме данных с использованием Matplotlib (Python ⇐ Python

[Anonymous]

У меня есть фрейм данных с тысячами значений глубины (y) и соответствующими значениями пористости (x) и именем слоя (layer). Я создал точечную диаграмму значений y и x и хочу раскрасить символы по слоям. Простой пример ниже:

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

#Import Libraries
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

#Create Data
data = {'y': [1,2,3,4,5,6,7,8,9,10],
'x': [1,2,3,4,5,6,7,8,9,10],
'layer' : ['a','a','a','b','b','b','c','c','c','c']}
df=pd.DataFrame(data)

print(df)

#Plot Data
plt.scatter(df['x'],df['y'])
plt.show()

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

y   x layer

0 1 1 а
1 2 2 а
2 3 3 а
3 4 4 б
4 5 5 б
5 6 6 b
6 7 7 c
7 8 8 c
8 9 9 c
9 10 10 c


Мне нужен простой способ раскрасить символы с помощью слой и создайте легенду.
Мне удалось раскрасить символы несколькими методами, например, добавив атрибут цвета в фрейм данных с помощью слой. Однако мне не удалось создать хорошую легенду для цветовых символов имени слоя. Любые предложения будут оценены.

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