จะเลือกสีใหม่สำหรับแต่ละเส้นที่พล็อตภายในรูปใน matplotlib ได้อย่างไร?

ฉันไม่ต้องการระบุสีสำหรับแต่ละเส้นพล็อตและให้แต่ละบรรทัดมีสีที่แตกต่างกัน แต่ถ้าฉันวิ่ง:

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

plt.show()

จากนั้นฉันจะได้ผลลัพธ์นี้:

หากคุณดูภาพด้านบนคุณจะเห็นว่า matplotlib พยายามเลือกสีสำหรับแต่ละบรรทัดที่แตกต่างกัน แต่ในที่สุดก็ใช้สีซ้ำ - สิบบรรทัดบนสุดใช้สีเดียวกันกับสิบอันดับล่าง ฉันแค่ต้องการหยุดไม่ให้ใช้สีที่ใช้แล้วซ้ำและ / หรือป้อนรายการสีที่จะใช้

matplotlib 1.5+

คุณสามารถใช้axes.set_prop_cycle( ตัวอย่าง )

matplotlib 1.0-1.4

คุณสามารถใช้axes.set_color_cycle( ตัวอย่าง )

matplotlib 0.x

คุณสามารถใช้Axes.set_default_color_cycle.

ฉันมักจะใช้อันที่สองต่อไปนี้:

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)

ตัวอย่าง 2:

prop_cycle

color_cycleถูกเลิกใช้ใน 1.5 เพื่อสนับสนุนลักษณะทั่วไปนี้: 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()

ยังแสดงในตัวอย่าง (ตอนนี้ชื่อไม่ดี): http://matplotlib.org/1.5.1/examples/color/color_cycle_demo.html ที่กล่าวถึงที่: https://stackoverflow.com/a/4971431/895245

ทดสอบใน matplotlib 1.5.1

ฉันไม่รู้ว่าคุณสามารถเปลี่ยนสีโดยอัตโนมัติได้หรือไม่ แต่คุณสามารถใช้ประโยชน์จากลูปของคุณเพื่อสร้างสีต่างๆได้:

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

ในกรณีนี้สีจะแตกต่างกันไปตั้งแต่สีดำจนถึงสีเขียว 100% แต่คุณสามารถปรับแต่งได้หากต้องการ

ดูเอกสารmatplotlib plot ()และค้นหาcolorอาร์กิวเมนต์คำหลัก

หากคุณต้องการป้อนรายการสีเพียงตรวจสอบให้แน่ใจว่าคุณมีรายการที่ใหญ่พอจากนั้นใช้ดัชนีของลูปเพื่อเลือกสี

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

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

คุณสามารถใช้ "colormap เชิงคุณภาพ" ที่กำหนดไว้ล่วงหน้าดังนี้:

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)

ทดสอบกับ matplotlib 3.0.3 ดูhttps://github.com/matplotlib/matplotlib/issues/10840axes.set_prop_cycle(color=cmap)สำหรับการอภิปรายเกี่ยวกับเหตุผลที่คุณไม่สามารถเรียก

รายการ colormaps เชิงคุณภาพที่กำหนดไว้ล่วงหน้ามีอยู่ที่https://matplotlib.org/gallery/color/colormap_reference.html :

คุณยังสามารถเปลี่ยนรอบสีเริ่มต้นในmatplotlibrcไฟล์ของคุณได้ หากคุณไม่ทราบว่าไฟล์นั้นอยู่ที่ไหนให้ทำดังต่อไปนี้ใน python:

import matplotlib
matplotlib.matplotlib_fname()

ซึ่งจะแสดงเส้นทางไปยังไฟล์ matplotlibrc ที่คุณใช้อยู่ในปัจจุบัน ในไฟล์นั้นคุณจะพบกับการตั้งค่าอื่น ๆ อีกมากมายเช่นaxes.color.cycleกัน เพียงใส่ลำดับสีที่คุณต้องการแล้วคุณจะพบมันในทุกพล็อตที่คุณทำ โปรดทราบว่าคุณยังสามารถใช้ชื่อสี html ที่ถูกต้องทั้งหมดใน matplotlib

ในฐานะบันทึกคำตอบของ Ciroคุณสามารถใช้prop_cycleเพื่อกำหนดรายการสีสำหรับ matplotlib เพื่อหมุนเวียน แต่จะมีกี่สี? จะเป็นอย่างไรหากคุณต้องการใช้วัฏจักรสีเดียวกันสำหรับพล็อตจำนวนมากโดยมีจำนวนเส้นต่างกัน

กลยุทธ์อย่างหนึ่งคือการใช้สูตรเช่นเดียวกับที่มาจากhttps://gamedev.stackexchange.com/a/46469/22397เพื่อสร้างลำดับสีที่ไม่มีที่สิ้นสุดโดยแต่ละสีพยายามให้แตกต่างอย่างมีนัยสำคัญจากทุกสีที่อยู่ข้างหน้า

น่าเสียดายที่prop_cycleไม่ยอมรับลำดับที่ไม่มีที่สิ้นสุด - มันจะค้างตลอดไปถ้าคุณผ่านมันไป แต่เราสามารถพูด 1,000 สีแรกที่สร้างจากลำดับดังกล่าวและตั้งเป็นวงจรสี ด้วยวิธีนี้สำหรับพล็อตที่มีจำนวนเส้นที่เหมาะสมคุณควรได้รับสีที่แตกต่างกัน

ตัวอย่าง:

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()

เอาท์พุต:

ตอนนี้สีทั้งหมดแตกต่างกัน - แม้ว่าฉันจะยอมรับว่าฉันพยายามแยกแยะไม่กี่สีก็ตาม!