最近有個需求,是做兩個數組重復程度計算,麻煩就麻煩在單個數組的元素有可能重復,處理思路如下:
1. 找到重復元素
2. 元素個數統計,利用np.bincount轉換,即元素個數統計到元素轉化的索引
3. 統計相同元素匹配個數
具體代碼如下:
# arr1, arr2都是np.array類型# 找到重復元素(交集)inters = np.intersect1d(arr1, arr2)# 元素個數索引轉換bc1 = np.bincount(arr1)bc2 = np.bincount(arr2)# 統計相同元素匹配個數same_count_list = [min(bc1[x], bc2[x]) for x in inters]same_count = sum(same_count_list)
以上這篇numpy 計算兩個數組重復程度的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。
新聞熱點
疑難解答