如何有效地获取numpy数组中相同值的数量？

如何解决如何有效地获取numpy数组中相同值的数量？

我有3D医学图像。我使用scipy.ndimage.measurements.label获取连接的体素组。非常快。但我也想获取每个标签的体素数量。我使用以下代码获取每个值的数量（假设image_3d是scipy.ndimage.measurements.label之后的数组）。花了大约2分钟。

import numpy as np
from skimage.measure import label
import time

image_3d = np.random.randint(100,size=(512,512,1024))
t1 = time.time()
pixel_count_list = [np.sum((image_3d== i).astype(int)) for i in range(100)]
t2 = time.time()
print("used time: ",t2-t1)
# 117 seconds

有什么有效的方法可以有效地获得它吗？

解决方法

import numpy as np
from skimage.measure import label
import time

image_3d = np.random.randint(100,size=(512,512,1024))
t1 = time.time()

unique_label,count_list = np.unique(image_3d,return_counts=True)
t2 = time.time()
print("used time: ",t2-t1)
# 2 seconds