使用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>中值类型即可。