I have a list of BigDecimals.
List<BigDecimal> amounts = new ArrayList<>()
How do i get the summary statistics of the above list using Java 8 streams without losing precision of upto 3-4 decimal places of the BigDecimal?
I created a BigDecimal
specialization of the generic summary statistics collector of this answer, which allowed extending it to also support summing, hence also calculating an average:
/**
* Like {@code DoubleSummaryStatistics}, {@code IntSummaryStatistics}, and
* {@code LongSummaryStatistics}, but for {@link BigDecimal}.
*/
public class BigDecimalSummaryStatistics implements Consumer<BigDecimal> {
public static Collector<BigDecimal,?,BigDecimalSummaryStatistics> statistics() {
return Collector.of(BigDecimalSummaryStatistics::new,
BigDecimalSummaryStatistics::accept, BigDecimalSummaryStatistics::merge);
}
private BigDecimal sum = BigDecimal.ZERO, min, max;
private long count;
public void accept(BigDecimal t) {
if(count == 0) {
Objects.requireNonNull(t);
count = 1;
sum = t;
min = t;
max = t;
}
else {
sum = sum.add(t);
if(min.compareTo(t) > 0) min = t;
if(max.compareTo(t) < 0) max = t;
count++;
}
}
public BigDecimalSummaryStatistics merge(BigDecimalSummaryStatistics s) {
if(s.count > 0) {
if(count == 0) {
count = s.count;
sum = s.sum;
min = s.min;
max = s.max;
}
else {
sum = sum.add(s.sum);
if(min.compareTo(s.min) > 0) min = s.min;
if(max.compareTo(s.max) < 0) max = s.max;
count += s.count;
}
}
return this;
}
public long getCount() {
return count;
}
public BigDecimal getSum()
{
return sum;
}
public BigDecimal getAverage(MathContext mc)
{
return count < 2? sum: sum.divide(BigDecimal.valueOf(count), mc);
}
public BigDecimal getMin() {
return min;
}
public BigDecimal getMax() {
return max;
}
@Override
public String toString() {
return count == 0? "empty": (count+" elements between "+min+" and "+max+", sum="+sum);
}
}
It can be used similar to the DoubleSummaryStatistics
counterpart, like
BigDecimalSummaryStatistics bds = list.stream().collect(BigDecimalSummaryStatistics.statistics());
As a full example:
List<BigDecimal> list = Arrays.asList(BigDecimal.ZERO, BigDecimal.valueOf(-2), BigDecimal.ONE);
BigDecimalSummaryStatistics bds = list.stream().collect(BigDecimalSummaryStatistics.statistics());
System.out.println(bds);
System.out.println("average: "+bds.getAverage(MathContext.DECIMAL128));
3 elements between -2 and 1, sum=-1
average: -0.3333333333333333333333333333333333
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With