Bagaimana cara memilih warna baru untuk setiap garis plot dalam gambar di matplotlib?

Saya TIDAK ingin menentukan warna untuk setiap garis yang diplot, dan meminta setiap baris mendapatkan warna yang berbeda. Tetapi jika saya menjalankan:

from matplotlib import pyplot as plt
for i in range(20):
    plt.plot([0, 1], [i, i])

plt.show()

maka saya mendapatkan output ini:

Jika Anda melihat gambar di atas, Anda dapat melihat bahwa matplotlib mencoba memilih warna untuk setiap baris yang berbeda, tetapi pada akhirnya ia menggunakan kembali warna - sepuluh baris teratas menggunakan warna yang sama dengan sepuluh baris terbawah. Saya hanya ingin menghentikannya mengulangi warna yang sudah digunakan DAN / ATAU memberinya daftar warna yang akan digunakan.

matplotlib 1.5+

Anda dapat menggunakan axes.set_prop_cycle( contoh ).

matplotlib 1.0-1.4

Anda dapat menggunakan axes.set_color_cycle( contoh ).

matplotlib 0.x

Anda bisa menggunakan Axes.set_default_color_cycle.

Saya biasanya menggunakan yang kedua dari ini:

from matplotlib.pyplot import cm
import numpy as np

#variable n below should be number of curves to plot

#version 1:

color=cm.rainbow(np.linspace(0,1,n))
for i,c in zip(range(n),color):
   plt.plot(x, y,c=c)

#or version 2:

color=iter(cm.rainbow(np.linspace(0,1,n)))
for i in range(n):
   c=next(color)
   plt.plot(x, y,c=c)

Contoh 2:

prop_cycle

color_cycletidak digunakan lagi di 1.5 karena mendukung generalisasi ini: http://matplotlib.org/users/whats_new.html#added-axes-prop-cycle-key-to-rcparams

# cycler is a separate package extracted from matplotlib.
from cycler import cycler
import matplotlib.pyplot as plt

plt.rc('axes', prop_cycle=(cycler('color', ['r', 'g', 'b'])))
plt.plot([1, 2])
plt.plot([2, 3])
plt.plot([3, 4])
plt.plot([4, 5])
plt.plot([5, 6])
plt.show()

Juga ditampilkan dalam contoh (sekarang bernama buruk): http://matplotlib.org/1.5.1/examples/color/color_cycle_demo.html disebutkan di: https://stackoverflow.com/a/4971431/895245

Diuji di matplotlib 1.5.1.

Saya tidak tahu apakah Anda dapat mengubah warna secara otomatis, tetapi Anda dapat memanfaatkan loop Anda untuk menghasilkan warna yang berbeda:

for i in range(20):
   ax1.plot(x, y, color = (0, i / 20.0, 0, 1)

Dalam hal ini, warna akan bervariasi dari hitam hingga 100% hijau, tetapi Anda dapat menyesuaikannya jika mau.

Lihat dokumen matplotlib plot () dan cari colorargumen kata kunci.

Jika Anda ingin memberi makan daftar warna, pastikan Anda memiliki daftar yang cukup besar dan kemudian gunakan indeks loop untuk memilih warna

colors = ['r', 'b', ...., 'w']

for i in range(20):
   ax1.plot(x, y, color = colors[i])

Anda dapat menggunakan "peta warna kualitatif" yang telah ditentukan sebelumnya seperti ini:

from matplotlib.cm import get_cmap

name = "Accent"
cmap = get_cmap(name)  # type: matplotlib.colors.ListedColormap
colors = cmap.colors  # type: list
axes.set_prop_cycle(color=colors)

Diuji pada matplotlib 3.0.3. Lihat https://github.com/matplotlib/matplotlib/issues/10840 untuk diskusi tentang mengapa Anda tidak dapat menelepon axes.set_prop_cycle(color=cmap).

Daftar peta warna kualitatif yang telah ditentukan tersedia di https://matplotlib.org/gallery/color/colormap_reference.html :

Anda juga dapat mengubah siklus warna default di matplotlibrcfile Anda . Jika Anda tidak tahu di mana file itu, lakukan hal berikut dengan python:

import matplotlib
matplotlib.matplotlib_fname()

Ini akan menunjukkan kepada Anda jalur ke file matplotlibrc yang saat ini Anda gunakan. Dalam file itu Anda akan menemukan di antara banyak pengaturan lain juga yang untuk axes.color.cycle. Masukkan saja urutan warna yang Anda inginkan dan Anda akan menemukannya di setiap plot yang Anda buat. Perhatikan bahwa Anda juga dapat menggunakan semua nama warna html yang valid di matplotlib.

Sebagai catatan jawaban Ciro , Anda dapat menggunakan prop_cycleuntuk mengatur daftar warna untuk matplotlib untuk digilir. Tapi berapa warnanya? Bagaimana jika Anda ingin menggunakan siklus warna yang sama untuk banyak plot, dengan jumlah garis yang berbeda?

Salah satu taktiknya adalah menggunakan rumus seperti yang ada di https://gamedev.stackexchange.com/a/46469/22397 , untuk menghasilkan urutan warna yang tak terbatas di mana setiap warna mencoba menjadi berbeda secara signifikan dari semua yang sebelumnya.

Sayangnya, prop_cycle tidak akan menerima urutan tak terbatas - itu akan menggantung selamanya jika Anda melewatinya. Tapi kita bisa mengambil, katakanlah, 1000 warna pertama yang dihasilkan dari urutan seperti itu, dan mengaturnya sebagai siklus warna. Dengan begitu, untuk plot dengan jumlah garis yang wajar, Anda akan mendapatkan warna yang dapat dibedakan.

Contoh:

from matplotlib import pyplot as plt
from matplotlib.colors import hsv_to_rgb
from cycler import cycler

# 1000 distinct colors:
colors = [hsv_to_rgb([(i * 0.618033988749895) % 1.0, 1, 1])
          for i in range(1000)]
plt.rc('axes', prop_cycle=(cycler('color', colors)))

for i in range(20):
    plt.plot([1, 0], [i, i])

plt.show()

Keluaran:

Sekarang, semua warna berbeda - meskipun saya akui bahwa saya kesulitan membedakan beberapa di antaranya!