Matplotlib scatter plot legend

Saya membuat sebuah 4D scatter plot grafik untuk mewakili suhu yang berbeda di daerah tertentu. Ketika saya membuat legenda, legenda menunjukkan simbol yang benar dan warna, tetapi menambahkan garis melalui itu. Kode I'm menggunakan adalah:

colors=['b', 'c', 'y', 'm', 'r']
lo = plt.Line2D(range(10), range(10), marker='x', color=colors[0])
ll = plt.Line2D(range(10), range(10), marker='o', color=colors[0])
l = plt.Line2D(range(10), range(10), marker='o',color=colors[1])
a = plt.Line2D(range(10), range(10), marker='o',color=colors[2])
h = plt.Line2D(range(10), range(10), marker='o',color=colors[3])
hh = plt.Line2D(range(10), range(10), marker='o',color=colors[4])
ho = plt.Line2D(range(10), range(10), marker='x', color=colors[4])
plt.legend((lo,ll,l,a, h, hh, ho),('Low Outlier', 'LoLo','Lo', 'Average', 'Hi', 'HiHi', 'High Outlier'),numpoints=1, loc='lower left', ncol=3, fontsize=8)

Saya mencoba mengubah Line2D untuk Menyebarkan dan menyebar. Menyebarkan kembali kesalahan dan menyebarkan berubah grafik dan kembali kesalahan.

Dengan menyebarkan, aku berubah range(10) ke daftar yang berisi titik-titik data. Setiap daftar berisi baik x, y, atau z variabel.

lo = plt.scatter(xLOutlier, yLOutlier, zLOutlier, marker='x', color=colors[0])
ll = plt.scatter(xLoLo, yLoLo, zLoLo, marker='o', color=colors[0])
l = plt.scatter(xLo, yLo, zLo, marker='o',color=colors[1])
a = plt.scatter(xAverage, yAverage, zAverage, marker='o',color=colors[2])
h = plt.scatter(xHi, yHi, zHi, marker='o',color=colors[3])
hh = plt.scatter(xHiHi, yHiHi, zHiHi, marker='o',color=colors[4])
ho = plt.scatter(xHOutlier, yHOutlier, zHOutlier, marker='x', color=colors[4])
plt.legend((lo,ll,l,a, h, hh, ho),('Low Outlier', 'LoLo','Lo', 'Average', 'Hi', 'HiHi',     'High Outlier'),scatterpoints=1, loc='lower left', ncol=3, fontsize=8)

Ketika saya menjalankan ini, legenda tidak ada lagi, itu adalah sebuah kotak putih kecil di sudut dengan apa-apa di dalamnya.

Ada saran?

sodd
Pertanyaan edit 21 Januari 2016 в 2:09
Pemrograman
python matplotlib scatter-plot legend
1 Juli 2013 в 7:12
10 tayangan
Mengomentari pertanyaan (1)
Larutan

2D scatter plot

Menggunakan menyebarkan metode matplotlib.pyplot modul harus bekerja (setidaknya dengan matplotlib 1.2.1 dengan Python 2.7.5), seperti pada contoh kode di bawah ini. Juga, jika anda menggunakan scatter plot, gunakan scatterpoints=1 daripada numpoints=1 legenda panggilan untuk hanya memiliki satu titik untuk setiap legenda entri.

Dalam kode di bawah ini saya've digunakan nilai acak daripada merencanakan kisaran yang sama berulang-ulang, membuat semua plot terlihat (yaitu tidak tumpang tindih satu sama lain).

import matplotlib.pyplot as plt
from numpy.random import random

colors = ['b', 'c', 'y', 'm', 'r']

lo = plt.scatter(random(10), random(10), marker='x', color=colors[0])
ll = plt.scatter(random(10), random(10), marker='o', color=colors[0])
l  = plt.scatter(random(10), random(10), marker='o', color=colors[1])
a  = plt.scatter(random(10), random(10), marker='o', color=colors[2])
h  = plt.scatter(random(10), random(10), marker='o', color=colors[3])
hh = plt.scatter(random(10), random(10), marker='o', color=colors[4])
ho = plt.scatter(random(10), random(10), marker='x', color=colors[4])

plt.legend((lo, ll, l, a, h, hh, ho),
           ('Low Outlier', 'LoLo', 'Lo', 'Average', 'Hi', 'HiHi', 'High Outlier'),
           scatterpoints=1,
           loc='lower left',
           ncol=3,
           fontsize=8)

plt.show()

3D scatter plot

Untuk plot pencar di 3D, gunakan plot metode, seperti legenda yang tidak mendukung Patch3DCollection sebagai dikembalikan oleh menyebarkan metode Axes3D misalnya. Untuk menentukan markerstyle anda dapat memasukkan ini sebagai posisi argumen dalam metode panggilan, seperti yang terlihat pada contoh di bawah ini. Opsional satu dapat memiliki argumen untuk kedua linestyle dan penanda parameter.

import matplotlib.pyplot as plt
from numpy.random import random
from mpl_toolkits.mplot3d import Axes3D

colors=['b', 'c', 'y', 'm', 'r']

ax = plt.subplot(111, projection='3d')

ax.plot(random(10), random(10), random(10), 'x', color=colors[0], label='Low Outlier')
ax.plot(random(10), random(10), random(10), 'o', color=colors[0], label='LoLo')
ax.plot(random(10), random(10), random(10), 'o', color=colors[1], label='Lo')
ax.plot(random(10), random(10), random(10), 'o', color=colors[2], label='Average')
ax.plot(random(10), random(10), random(10), 'o', color=colors[3], label='Hi')
ax.plot(random(10), random(10), random(10), 'o', color=colors[4], label='HiHi')
ax.plot(random(10), random(10), random(10), 'x', color=colors[4], label='High Outlier')

plt.legend(loc='upper left', numpoints=1, ncol=3, fontsize=8, bbox_to_anchor=(0, 0))

plt.show()

sodd
Jawaban edit 2 Juli 2013 в 2:44
Komentar (2)

Berikut ini's cara yang lebih mudah untuk melakukan hal ini (sumber: di sini):

import matplotlib.pyplot as plt
from numpy.random import rand

fig, ax = plt.subplots()
for color in ['red', 'green', 'blue']:
    n = 750
    x, y = rand(2, n)
    scale = 200.0 * rand(n)
    ax.scatter(x, y, c=color, s=scale, label=color,
               alpha=0.3, edgecolors='none')

ax.legend()
ax.grid(True)

plt.show()

Dan anda'akan mendapatkan ini:

Lihat di sini untuk legenda sifat

Amir
Jawaban edit 9 April 2019 в 7:51
Komentar (0)

jika anda menggunakan matplotlib versi 3.1.1 atau di atas, anda dapat mencoba:

import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap

x = [1, 3, 4, 6, 7, 9]
y = [0, 0, 5, 8, 8, 8]
classes = ['A', 'B', 'C']
values = [0, 0, 1, 2, 2, 2]
colours = ListedColormap(['r','b','g'])
scatter = plt.scatter(x, y,c=values, cmap=colours)
plt.legend(handles=scatter.legend_elements()[0], labels=classes)

Komentar (0)

Jawaban yang lain tampak sedikit rumit, anda hanya dapat menambahkan sebuah parameter 'label' dalam menyebarkan fungsi dan yang akan menjadi legenda untuk rencana anda.

import matplotlib.pyplot as plt
from numpy.random import random

colors = ['b', 'c', 'y', 'm', 'r']

lo = plt.scatter(random(10), random(10), marker='x', color=colors[0],label='Low Outlier')
ll = plt.scatter(random(10), random(10), marker='o', color=colors[0],label='LoLo')
l  = plt.scatter(random(10), random(10), marker='o', color=colors[1],label='Lo')
a  = plt.scatter(random(10), random(10), marker='o', color=colors[2],label='Average')
h  = plt.scatter(random(10), random(10), marker='o', color=colors[3],label='Hi')
hh = plt.scatter(random(10), random(10), marker='o', color=colors[4],label='HiHi')
ho = plt.scatter(random(10), random(10), marker='x', color=colors[4],label='High Outlier')

plt.legend(loc='upper center', bbox_to_anchor=(0.5, -0.05),
          fancybox=True, shadow=True, ncol=4)

plt.show()

Berikut ini adalah output

Komentar (0)