我正在使用 CIFAR 10 資料集，并希望創建一個函式來回傳每個類的影像數量。我有一個字典 classes_dict ，它將類名映射到值，見下文：
classes_dict= {'airplane': 0,
'automobile': 1,
'bird': 2,
'cat': 3,
'deer': 4,
'dog': 5,
'frog': 6,
'horse': 7,
'ship': 8,
'truck': 9}
還有一個字典 images_per_class 計算每個類的頻率：
from collections import Counter
img_per_class = Counter(y_train)
img_per_class
Counter({0: 5000,
1: 5000,
2: 5000,
3: 5000,
4: 5000,
5: 5000,
6: 5000,
7: 5000,
8: 5000,
9: 5000})
我想更改 images_per_class 中的鍵，使其對應于正確的類名。期望的輸出：
{'airplane': 5000,
'automobile': 5000,
'bird': 5000,
'cat': 5000,
'deer': 5000,
'dog': 5000,
'frog': 5000,
'horse': 5000,
'ship': 5000,
'truck': 5000}
uj5u.com熱心網友回復：
您可以使用 dict-comprehension。
result = {k: img_per_class[v] for k, v in classes_dict.items()}
Counter（如果你必須有一個，你可以將此 dict 傳遞給建構式Counter。）
轉載請註明出處，本文鏈接：https://www.uj5u.com/net/441344.html
上一篇：收到錯誤時如何使用git提交：等待您的編輯器關閉檔案...？
下一篇：地圖中的常規輸出