数据集是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