У меня есть следующий фрагмент кода (значения для c, s, x, y являются макетами, но настоящие списки имеют тот же формат, только намного больше. Используются только два цвета - красный и зеленый. Все списки одинаковы размер)

Проблема в том, что цветовая легенда не материализуется. Я совершенно не понимаю, почему. Фрагменты кода для создания легенды - это в основном вырезанные и вставленные из документов, т.е. (https://matplotlib.org/3.1.1/gallery/lines_bars_and_markers/scatter_with_legend.html#sphx-glr-gallery-lines- полосы-и-маркеры-разброс-с-легендой-ру)

У кого-нибудь есть идеи ??

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
%matplotlib inline

c = [ 'g', 'r', 'r', 'g', 'g', 'r', 'r', 'r', 'g', 'r']
s = [ 10, 20, 10, 40, 60, 90, 90, 50, 60, 40]
x = [ 2.4, 3.0, 3.5, 3.5, 3.5, 3.5, 3.5, 2.4, 3.5, 3.5]
y = [24.0, 26.0, 20.0, 19.0, 19.0, 21.0, 20.0, 23.0, 20.0, 20.0]

fig, ax = plt.subplots()

scatter = plt.scatter(x, y, s=s, c=c, alpha=0.5)

# produce a legend with the unique colors from the scatter
handles, lables = scatter.legend_elements()
legend1 = ax.legend(handles, labels, loc="lower left", title="Colors")
ax.add_artist(legend1)

# produce a legend with a cross section of sizes from the scatter
handles, labels = scatter.legend_elements(prop="sizes", alpha=0.5)
legend2 = ax.legend(handles, labels, loc="upper right", ncol=2, title="Sizes")

plt.show()

Вывод сюжета:

enter image description here

0
John Durand 28 Май 2020 в 23:37

1 ответ

Лучший ответ

Кажется, что legend_elements() предназначен для использования только тогда, когда c= передается числовой массив для сопоставления с цветовой картой. Вы можете протестировать, заменив c=c на c=s в своем коде, и вы получите желаемый результат.

Лично я ожидал, что ваш код будет работать, и, возможно, стоит рассказать об этом либо как об ошибке, либо как о запросе функции на github для matplotlib. РЕДАКТИРОВАТЬ : на самом деле, эта проблема уже обсуждается на средство отслеживания проблем

Один из способов обойти это ограничение - заменить массив имен цветов числовым массивом и создать настраиваемую цветовую карту, которая сопоставляет каждое значение в вашем массиве желаемому цвету:

#c = [ 'g', 'r', 'r', 'g', 'g', 'r', 'r', 'r', 'g', 'r']
c = [0, 1, 1, 0, 0, 1, 1, 1, 0, 1]
cmap = matplotlib.colors.ListedColormap(['g','r'])
s = [ 10, 20, 10, 40, 60, 90, 90, 50, 60, 40]
x = [ 2.4, 3.0, 3.5, 3.5, 3.5, 3.5, 3.5, 2.4, 3.5, 3.5]
y = [24.0, 26.0, 20.0, 19.0, 19.0, 21.0, 20.0, 23.0, 20.0, 20.0]

fig, ax = plt.subplots()

scatter = plt.scatter(x, y, s=s, c=c, alpha=0.5, cmap=cmap)

# produce a legend with the unique colors from the scatter
handles, labels = scatter.legend_elements()
legend1 = ax.legend(handles, labels, loc="lower left", title="Colors")
ax.add_artist(legend1)

# produce a legend with a cross section of sizes from the scatter
handles, labels = scatter.legend_elements(prop="sizes", alpha=0.5)
legend2 = ax.legend(handles, labels, loc="upper right", ncol=2, title="Sizes")

plt.show()

enter image description here

1
Diziet Asahi 28 Май 2020 в 21:54