java - 使用 java 流将评级列表转换为平均评级列表

Question

我在 Java 中遇到了流的问题，这困扰了我一整天。因此，如果有人可以帮助我，我会非常高兴。

我有两个类(class) Rate 和 AvgRate:

class Rate {
    private int itemId;
    private int userId;
    private int rate;

    [...]
}

class AvgRate {
    private int itemId;
    private int avgRate;

    [...]
}

假设我有费率列表:

//Rate(itemId, userId, rate)
List<Rate> rates = Arrays.asList(
    new Rate(1, 1, 3),
    new Rate(1, 2, 1),
    new Rate(1, 3, 4),
    new Rate(1, 4, 2),
    new Rate(2, 1, 5),
    new Rate(2, 2, 1),
    new Rate(3, 3, 2),
);

现在我想计算每个项目的平均评分并将其放入列表中，例如

//(itemId, avgRating)
List<AvgRate> avgRates = {
    (1, 2,5),
    (2, 3),
    (3, 2)
}

所以问题是我如何使用 Java 流来执行此操作。

感谢您的建议。

Answer 1

尝试这样的事情:

Map<Integer, Integer> avgInteger = rates.stream()
  .collect(Collectors.groupingBy(Rate::getItemId, Collectors.averagingInt(Rate::getRate));

List<AvgRate> avgRates = avgInteger.entrySet().stream()
  .map(entry -> new AvgRate(entry.getKey(), entry.getValue()))
  .collect(Collectors.toList());

第一个 block 将您的列表转换为 itemId 和 rate 平均值的 Map。第二个 block 将 Map 转换回 List。