用于为绘图 python 分配颜色的循环

Question

我正在使用 for 循环为类分配数字，我在这方面取得了成功，但我发现很难根据使用相同 for 循环的类的数量同时插入不同的颜色，但我收到错误： 'list'对象不能解释为整数

下面是代码：

n_class = 5
colors = ['r', 'g', 'b', 'y','k', 'y']
# plotting
for i, c in range(n_class, colors):
    plt.plot(fpr[i], tpr[i], linestyle='--',color=[c], label= 'Class %d' %i )
        

plt.title('Multiclass ROC curve')
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive rate')
plt.legend(loc='best')
plt.savefig('Multiclass ROC',dpi=300);

Answer 1

错误在这一行

for i, c in range(n_class, colors):

很简单，你要做的就是得到列表的长度

for i, c in range(n_class, len(colors)):

但我不认为这是主要的错误，因为那样它就会把它当作

range(5,6)

如果你想给所有的图一个你需要的颜色

for i in range(len(colors)):
    plt.plot(fpr[i], tpr[i], linestyle='--',color=colors[i], label= 'Class %d' %i )