Java 对数据分组

Posted 2023-05-13

有这样两组数据，每一行数据是一次交易的记录，现在想把数据分组，分组原则是这样的：将行与行之间有重合的元素并且之间有相同元素的放在一个集合中，有几组有重合元素且相同的元素，就放在几个集合中。没有重合的元素分别放在不同的集合中。如上图，最终的结果应该是这样的：
[8017, 8002, 8013, 1087, 8015, 8016, 8012, 8014]
[5409, 5022, 5011]
[8010, 8003]

参考技术A 把从数据库读出来的ResultSet 传入这个方法

public static Map<Long, List<Long>> orgnaizeData(ResultSet rs)
Map<Long, List<Long>> result = null;
try
if (rs.getRow() > 0)
result = new HashMap<Long, List<Long>>();
while (rs.next())
Long key = rs.getLong(1);
Long value = rs.getLong(2);
if (result.containsKey(key))
result.get(key).add(value);
else
List<Long> valueList = new ArrayList<Long>();
valueList.add(value);
result.put(key, valueList);



catch (SQLException e)
// TODO Auto-generated catch block
e.printStackTrace();

return result;
参考技术B Map<Integer,List<Integer>> map
用这个 数据结构 搞定他。 参考技术C map

java~集合分组groupby的实现

对于数据聚合来说，分组操作是很常见的，在.net里有lambda和linq，而在java里也有lambda，现在我们来实现对一个集合进行分组。

一 准备工作，有两个类型

  @Value
  class Item {
    private Date createAt;
    private int count;
    private BigDecimal price;
  }

  @Value
  class Product {
    private String name;
    private String code;
    private List<Item> items;
  }

二 为两个类型建立集合，并赋值

    List<Product> products = new ArrayList<>();
    products.add(new Product("apple", "1001", Arrays.asList(
        new Item(new Date(2018, 1, 1), 10, new BigDecimal("9.99")))));

    products.add(new Product("apple", "1001", Arrays.asList(
        new Item(new Date(2018, 2, 1), 10, new BigDecimal("19.99")))));

    products.add(new Product("apple", "1001", Arrays.asList(
        new Item(new Date(2018, 3, 1), 10, new BigDecimal("29.99")))));

三 使用lambda进行分组，主要对name字段进行分组，然后把结果存在一个新的集合里

    Map<String, List<Product>> groupByPriceMap =
        products.stream().collect(Collectors.groupingBy(Product::getName));
    products = new ArrayList<>();
    for (Map.Entry<String, List<Product>> str : groupByPriceMap.entrySet()) {
      List<Item> items = new ArrayList<>();
      for (Product product : str.getValue()) {
        items.addAll(product.getItems());
      }
      products.add(new Product(str.getKey(), "", items));
    }

四 调试代码，在断点处查看分组后的结果

五 多条件分组的实现

  Function<Product, List<Object>> compositeKey = personRecord ->
        Arrays.asList(personRecord.getName(), personRecord.getCode());

    Map<Object, List<Product>> map =
        products.stream().collect(Collectors.groupingBy(compositeKey));

 感谢阅读！