Matplotlib基于x轴上的值绘制多种颜色

Question

我希望得到一个类似下图的图，它根据x轴的值有不同的颜色。忽略u和f字母以及蓝色曲线和灰色线条。我只需要绿色和红色线条。因此，如果您使用我的代码，您将获得一种颜色的图表。我想要的是当x在0和转折点之间（在这种情况下它是x=50%）时有不同的颜色，然后是其他颜色。

代码：

import matplotlib.pyplot as plt

def GRLC(values):
    n = len(values)
    assert(n > 0), 'Empty list of values'
    sortedValues = sorted(values) #Sort smallest to largest

    #Find cumulative totals
    cumm = [0]
    for i in range(n):
        cumm.append(sum(sortedValues[0:(i + 1)]))

    #Calculate Lorenz points
    LorenzPoints = [[], []]
    sumYs = 0           #Some of all y values
    robinHoodIdx = -1   #Robin Hood index max(x_i, y_i)
    for i in range(1, n + 2):
        x = 100.0 * (i - 1)/n
        y = 100.0 * (cumm[i - 1]/float(cumm[n]))
        LorenzPoints[0].append(x)
        LorenzPoints[1].append(y)
        sumYs += y
        maxX_Y = x - y
        if maxX_Y > robinHoodIdx: robinHoodIdx = maxX_Y   

    giniIdx = 100 + (100 - 2 * sumYs)/n #Gini index 

    return [giniIdx, giniIdx/100, robinHoodIdx, LorenzPoints]

reg=[400,200]
result_reg = GRLC(reg)
print 'Gini Index Reg', result_reg[0]  
print 'Gini Coefficient Reg', result_reg[1]
print 'Robin Hood Index Reg', result_reg[2]

#Plot
plt.plot(result_reg[3][0], result_reg[3][1], [0, 100], [0, 100], '--')

plt.legend(['Reg-ALSRank@10','Equity-Line'], loc='upper left',prop={'size':16})
plt.xlabel('% of items ')
plt.ylabel('% of times being recommended')
plt.show()

Answer 1

这就是你如何绘制两行不同颜色的方法，知道颜色应该改变的数组中的索引。

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0,49, num=50)
y = x**2

x0 = 23

plt.plot(x[:x0+1], y[:x0+1])
plt.plot(x[x0:], y[x0:])
plt.show()

这是有效的，因为默认情况下，后续的线条图有不同的颜色，但您当然可以自己设置颜色，

plt.plot(x[:x0+1], y[:x0+1], color="cornflowerblue")