[英]arg max in Java 8 streams?
根据产生双精度或整数值的标准的最大化,我经常需要集合的最大元素。 Streams 有 max() 函数,它需要我实现一个比较器,我觉得它很麻烦。 是否有更简洁的语法,例如以下示例中的names.stream().argmax(String::length)
?
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
public class ArgMax
{
public static void main(String[] args)
{
List<String> names = Arrays.asList("John","Joe","Marilyn");
String longestName = names.stream().max((String s,String t)->(Integer.compare(s.length(),t.length()))).get();
System.out.println(longestName);
}
}
用
String longestName = names.stream().max(Comparator.comparing(String::length)).get();
比较某些属性上的元素(可能比这更复杂,但不是必须的)。
正如 Brian 在评论中所建议的那样,如果Stream
可能为空,则像这样使用Optional#get()
是不安全的。 您更适合使用一种更安全的检索方法,例如Optional#orElse(Object)
,如果没有最大值,它将为您提供一些默认值。
我认为应该考虑的是,虽然max
/ min
是唯一的,但这当然不能保证argMax
/ argMin
; 这特别意味着减少的类型应该是一个集合,例如一个List
。 这需要比上面建议的更多的工作。
以下ArgMaxCollector<T>
类提供了这种减少的简单实现。 main
展示了该类用于计算字符串集的argMax
/ argMin
的应用程序
one two three four five six seven
按长度排序。 输出(分别报告argMax
和argMin
收集器的结果)应该是
[three, seven]
[one, two, six]
分别是最长的两个字符串和三个最短的字符串。
这是我第一次尝试使用新的 Java 8 流 API,因此非常欢迎任何评论!
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collector;
class ArgMaxCollector<T> {
private T max = null;
private ArrayList<T> argMax = new ArrayList<T>();
private Comparator<? super T> comparator;
private ArgMaxCollector( Comparator<? super T> comparator ) {
this.comparator = comparator;
}
public void accept( T element ) {
int cmp = max == null ? -1 : comparator.compare( max, element );
if ( cmp < 0 ) {
max = element;
argMax.clear();
argMax.add( element );
} else if ( cmp == 0 )
argMax.add( element );
}
public void combine( ArgMaxCollector<T> other ) {
int cmp = comparator.compare( max, other.max );
if ( cmp < 0 ) {
max = other.max;
argMax = other.argMax;
} else if ( cmp == 0 ) {
argMax.addAll( other.argMax );
}
}
public List<T> get() {
return argMax;
}
public static <T> Collector<T, ArgMaxCollector<T>, List<T>> collector( Comparator<? super T> comparator ) {
return Collector.of(
() -> new ArgMaxCollector<T>( comparator ),
( a, b ) -> a.accept( b ),
( a, b ) ->{ a.combine(b); return a; },
a -> a.get()
);
}
}
public class ArgMax {
public static void main( String[] args ) {
List<String> names = Arrays.asList( new String[] { "one", "two", "three", "four", "five", "six", "seven" } );
Collector<String, ArgMaxCollector<String>, List<String>> argMax = ArgMaxCollector.collector( Comparator.comparing( String::length ) );
Collector<String, ArgMaxCollector<String>, List<String>> argMin = ArgMaxCollector.collector( Comparator.comparing( String::length ).reversed() );
System.out.println( names.stream().collect( argMax ) );
System.out.println( names.stream().collect( argMin ) );
}
}
简单有效的解决方案在这里:
https://stackoverflow.com/a/63201174/2069400
/** return argmin item, else null if none. NAN scores are skipped */
public static <T> T argmin(Stream<T> stream, ToDoubleFunction<T> scorer) {
Double min = null;
T argmin = null;
for (T p: (Iterable<T>) stream::iterator) {
double score = scorer.applyAsDouble(p);
if (min==null || min > score) {
min = score;
argmin = p;
}
}
return argmin;
}
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.