- Counter()
from collections import Counter
y = torch.randint(0, 5, (20,)) # 20個(gè)sample的真實(shí)標(biāo)簽，值為0-4。
print(Counter(y.numpy()))
"""
Counter()會(huì)生成一個(gè)字典，讀出numpy array中各個(gè)類含有多少sample。
output:
Counter({3: 3, 2: 4, 4: 3, 0: 4, 1: 6})
"""
np.bincount()
np.bincount(y)
"""
np.bincount()會(huì)生成一個(gè)array，每一個(gè)元素代表一個(gè)類的samples的個(gè)數(shù)。
output:
array([5, 3, 3, 5, 4])
"""
- np.unique()
y = torch.randint(0, 5, (20,)) # 20個(gè)sample的真實(shí)標(biāo)簽，值為0-4。
print(np.unique(y))
"""
np.unique()生成一個(gè)array，輸出有多少類別。
output:
array([0, 1, 2, 3, 4])
"""
- 取出滿足條件的元素
# 去除0元素/取出非0元素
t = np.array([1,1,1,2,0,6,7,8,2,2])
print(t[t!=0])
"""
output:
[1 1 1 2 6 7 8 2 2]
"""
# 取出滿足條件的元素
a = np.array([0,-1,-2,5,7,18,9,-4])
print((a<0))
print(a[(a<0)])
print(a[a<0])
"""
output:
[False True True False False False False True]
[-1 -2 -4]
[-1 -2 -4]
"""
