使用Comparator根据Map的value进行排序

使用Comparator根据Map的value进行排序:

在写推荐系统时,使用到余弦相似度算法对用户进行推荐,里面需要根据value值对推荐由相似度高到底排序,所以撸了一个实现Comparator对map中value进行排序的工具方法。

 public static Map<String, Float> sortMapByValue(Map<String, Float> oriMap) {
        if (oriMap == null || oriMap.isEmpty()) {
            return null;
        }
        Map<String, Float> sortedMap = new LinkedHashMap<String, Float>();
        List<Map.Entry<String, Float>> entryList = new ArrayList<Map.Entry<String, Float>>(
                oriMap.entrySet());
        Collections.sort(entryList, new MapValueComparator());

        Iterator<Map.Entry<String, Float>> iter = entryList.iterator();
        Map.Entry<String, Float> tmpEntry = null;
        while (iter.hasNext()) {
            tmpEntry = iter.next();
            sortedMap.put(tmpEntry.getKey(), tmpEntry.getValue());
        }
        return sortedMap;
    }

    public static class MapValueComparator implements Comparator<Map.Entry<String, Float>> {

        @Override
        public int compare(Map.Entry<String, Float> v1, Map.Entry<String, Float> v2) {
            //按照从大到小的顺序排列,如果想正序 调换v1,v2的位置
            return v2.getValue().compareTo(v1.getValue());
        }
    }

根据value中float进行排序,如果需要从小到大,则修改v1和v2的位置即可。如果传入value为其他类型,修改 Map<String, Float>中值类型即可。

发布了41 篇原创文章 · 获赞 108 · 访问量 10万+

猜你喜欢

转载自blog.csdn.net/qq_37857921/article/details/103489923
今日推荐