數據集是kaggle的dog breed這個競賽中的標簽文件，名為'labels.csv'，前5行如下：

方法一：

class_df = pd.read_csv('labels.csv') class_df.head() class_to_idx = {x:i for i,x in enumerate(class_df.breed.unique())} idx_to_class = {i:x for i,x in enumerate(class_df.breed.unique())}

方法二：

df = pd.read_csv('labels.csv') codes = range(len(df.breed.unique()))#狗種類數 labelnames = df.breed.unique()#狗種類名 <class 'numpy.ndarray'> breed_to_code = dict(zip(labelnames, codes))#'otterhound': 19 code_to_breed = dict(zip(codes, labelnames))#0: 'boston_bull'

從結果可以看到unique()相當于把list變為set

與50位技術專家面對面20年技術見證，附贈技術全景圖

總結

以上是生活随笔為你收集整理的类别名转成数字标签的全部內容，希望文章能夠幫你解決所遇到的問題。

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