根据Java列表中的第一个元素进行筛选

Question

我想通过列表中的第一个元素进行过滤，然后通过第二个元素进行分组。

public class MyClass{
int index;
String fruit;
int quantity;
public MyClass(int index, String fruit, int quantity){
    this.index = index;
    this.fruit = fruit;
    this.quantity = quantity;
}

public int getIndex(){
    return index;
}

public String getFruit(){
    return fruit;
}

public int getQuantity(){
    return quantity;
}

ArrayList<MyClass> test = new ArrayList<MyClass>();

MyClass t1 = new MyClass(1, "apple", 6);
test.add(t1);
MyClass t2 = new MyClass(2, "apple", 6);
test.add(t2);
MyClass t3 = new MyClass(1, "banana", 6);
test.add(t3);
MyClass t4 = new MyClass(2, "banana", 6);
test.add(t4);
...
Myclass t20 = new MyClass(10, "apple", 6);

if (MyClass.getIndex() <= 5){
    Map<String, Integer> map = test.stream()
        .collect(groupingBy(MyClass::fruit, averagingLong(MyClass::quantity)));
}

//desired return 
// {apple: 12, banana:12}
}

在使用Java Stream获得平均值之前，我过滤了第一个元素，即index。这样做合适吗？

Answer 1

你说平均，但你的问题显示了以下内容：

//期望的回报
//{苹果：12,香蕉：12}

要提供所需的返回，您需要执行summingInt，而不是averagingInt

Map<String, Integer> map = test.stream().filter(t->t.getIndex()<=5)
        .collect(Collectors.groupingBy(MyClass::getFruit,
                Collectors.summingInt(MyClass::getQuantity)));
                
System.out.println(map);

{banana=12, apple=12}