歡迎點擊「算法與編程之美」↑關注我們！

本文首發于微信公眾號："算法與編程之美"，歡迎關注，及時了解更多此系列文章。

 

在做機器學習分類問題的時候，經常需要將標簽名轉換為對應的索引。什么是label_to_index問題？下面將通過一個案例來詳細說明。

 

假設現在需要完成一個圖像分類的任務，現有五種類型的花卉，分別是：['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips']，每個類別的花卉有圖像若干。在做具體的圖形分類任務之前，需要將這些標簽名轉換為對應的整型索引，這樣的轉換便于機器學習模型的后續處理。

 

我們的目標是建立一個字典類型的數據結構label_to_index，能夠實現給定一個標簽名，找到指定的數字索引。給大家一點時間，如果是你，如何快速的編碼實現？

 

首先按照字典序對所有的標簽名進行排序。

class_names = ['roses', 'sunflowers', 'daisy', 'dandelion', 'tulips'] class_names = sorted(class_names) # ['daisy', 'dandelion', 'roses', 'sunflowers', 

總結

以上是生活随笔為你收集整理的Python基础 | 快速实现label_to_index的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。