matplotlib从plt.plot命令中检索颜色

Question

调用后如何获取plt.plot使用的颜色？ 我不想提前指定颜色。

def plotStuff():

    lines = plt.plot(np.random.rand(500))
    color = lines.magic_thing_get_color
    plt.plot(np.random.rand(500),color = color,label = "_nolegend_" )

所以调用plotStuff两次会使用一种颜色第一次绘制2个东西，第二次调用它时使用不同的颜色。

Answer 1

您正在搜索的魔术功能是get_color()。但是，plot命令返回带有行对象的列表，因此，您必须在项目而不是列表本身上调用此函数。你的功能看起来像

import matplotlib.pyplot as plt
import numpy as np
def plotStuff():
    lines = plt.plot(np.random.rand(20))
    color = lines[0].get_color()
    plt.plot(np.random.rand(20),color = color,label = "_nolegend_" )
plotStuff()

创建像这样的情节
两条线具有相同的颜色。