數(shù)據(jù)集是kaggle的dog breed這個(gè)競賽中的標(biāo)簽文件，名為'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()))#狗種類數(shù) 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'

從結(jié)果可以看到unique()相當(dāng)于把list變?yōu)閟et

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。