Я пытаюсь построить график Matplotlib с цветами в соответствии с именами в панде DataFrame. При этом на графике x, y разные точки имени имеют разный цвет.

Dataframe :

    id  x   y   Names   
0   MAC004524   29.137983   11.864633   ACORN-M     
1   MAC004525   28.14       11.80       ACORN-M 
2   MAC004526   24.14       12.80       ACORN-C 
....

Код:

names = set(df['Names'])
colors = list(cmap(np.linspace(0, 1, len(names))))
df['color']=0
for a, c in zip(names, colors):
    mask = df.loc[df['Names'] == a]
    df.loc[mask, 'color'] = c
#but get an error here KeyError: "[('i', 'd') ('x',) ('y',) ('A', 'c', 'o', 'r', 'n')\n ('A', 'c', 'o', 'r', 'n', '_', 'g', 'r', 'o', 'u', 'p', 'e', 'd')\n ('c', 'o', 'l', 'o', 'r')] not in index"

Тогда я хотел бы построить

x = df['x']
y = df['y']
c= df['color']
plt.scatter(x, y, c=c, s=1)

Требуется df:

    id  x   y   Names    color
0   MAC004524   29.137983   11.864633   ACORN-M    [0.267004 0.004874 0.329415 1.      ] 
0
proximacentauri 22 Окт 2018 в 14:01

2 ответа

Лучший ответ

Вы смотрели на морские заговоры? Вы можете сразу создать сюжет из исходного кадра данных:

import seaborn as sns
sns.scatterplot(x='x', y='y', hue='Names', data=df)
1
Jondiedoop 22 Окт 2018 в 14:02

Проблема должна заключаться в попытке разместить список в ячейке. Как показано здесь, вам следует использовать .at вместо .loc.

Вероятно, не самый эффективный способ, но он выполняет свою работу:

for a, c in zip(names, colors):
    mask = df[df['Names'] == a].index
    for value in mask:
        df.at[value, 'color'] = c

Я добавил индекс и перебрал каждое его значение, чтобы заменить его указанным цветом, так как мне еще предстоит найти способ добавить более одного значения в индекс строки команды .at.

1
Márcio Coelho 22 Окт 2018 в 13:59
52927770