如何在Java中找到列表中筛选出的元素的平均值
我想知道某一特定分支机构在某一年的平均销售额。有一张清单,上面有所有的分店和销售年限。我尝试使用IntStream来筛选具有所需年份和分支的元素,但我找不到平均值,indexOfSaleInTheYear.size()中显示错误,并且我不确定平均部件代码是否正确。请帮帮忙。谢谢。
Scanner avgSale = new Scanner(System.in);
System.out.println("Please enter the year for average sale:
");
int yearEntered = avgSale.nextInt();
Scanner branchNameForAvg = new Scanner(System.in);
System.out.println("Please enter the branch name whose average sale is to be found, case sensitive:
");
String branchNameEntered = branchNameForAvg.nextLine();
Stream<Integer> indexOfSalesInTheYear = IntStream.of(0, Sale.allValues.size()).boxed()
.filter(i -> Sale.allYears.get(i).equals(yearEntered))
.filter(i -> Sale.allBranches.get(i).equals(branchNameEntered));
int sum = 0;
for(int j = 0; j < indexOfSalesInTheYear.size(); j++){
sum += j;
}
double avgSaleFinal = sum/indexOfSalesInTheYear.size();
System.out.println("The average sale for the year " + yearEntered + " is " + avgSaleFinal + ".");
}
解决方案
以下是我处理它的方式。不是单独的值列表、年份列表和分支列表,而是包含以下字段的对象列表:
record Sale(int value, int year, String branch) {
}
这里我使用的是record
,但如果您愿意,需要具有可变的Sale
实例,或者正在使用Java的早期版本,您可以调整此代码以使用具有普通getter和setter的类。
List<Sale> sales = List.of(
new Sale(12, 2020, "east"),
new Sale(15, 2020, "west"),
new Sale(13, 2020, "east"),
new Sale(16, 2020, "west"),
new Sale(14, 2020, "east"),
new Sale(17, 2020, "west"),
new Sale(18, 2021, "east"),
new Sale(19, 2021, "west"));
double averageOfSales(int year, String branchName) {
OptionalDouble average = sales.stream()
.filter(sale -> sale.year() == year)
.filter(sale -> sale.branch().equals(branchName))
.mapToInt(sale -> sale.value())
.average();
return average.orElseThrow(() ->
new NoSuchElementException("no matching year and branch"));
}
您可能只想返回OptionalDouble
或抛出一个不同的异常;这完全取决于您希望如何处理没有提供年份和分支机构名称的情况。在这种情况下,您甚至可能只想使用return 0.0
或NaN或-1.0或其他一些标记值。
由于IntStream
已有求和或求平均值的方法,因此无需自己对这些值求和或除以值的个数即可得出平均值。
相关文章