用於為繪圖 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 )