如何使用Stream API从列表中获取随机元素？

Question

如何使用Stream API从列表中获取随机元素？

31

使用Java8的流API从列表中获取随机元素的最有效方法是什么？

Arrays.asList(new Obj1(), new Obj2(), new Obj3());

谢谢。

- aekber

9个回答

7

有更有效的方法可以完成这个任务，但如果必须使用Stream，最简单的方法是创建自己的比较器，返回随机结果（-1、0、1），并对流进行排序：

 List<String> strings = Arrays.asList("a", "b", "c", "d", "e", "f");
    String randomString = strings
            .stream()
            .sorted((o1, o2) -> ThreadLocalRandom.current().nextInt(-1, 2))
            .findAny()
            .get();

ThreadLocalRandom提供了一个"开箱即用"的方法，可以在你需要的范围内获取随机数，以供比较使用。

- RichardK

5

不错的 hack，但它违反了 Comparator::compareTo 协定。 - Auberon

3

它违反了 Comparator::compareTo 的约定。 - Illya Kysil

3

打破Comparator::compareTo协定 - Shane

3

Comparator::compareTo合约 - sp00m

3

它坏了。 - Stephan Henningsen

显示剩余2条评论

7

虽然所有给出的答案都有效，但有一个简单的一行代码可实现此功能，无需首先检查列表是否为空：

List<String> list = List.of("a", "b", "c");
list.stream().skip((int) (list.size() * Math.random())).findAny();

对于一个空列表，这将返回一个Optional.empty。

- Mikusch

这应该是被接受的答案。我同意，你可以在这里使用非流式方法，但在我的情况下，第三方库已经返回了一个流，在那里我需要获取一个随机元素。 - Stefan Haberl

5

上一次我需要做这样的事情时，我是这样做的：

List<String> list = Arrays.asList("a", "b", "c");
Collections.shuffle(list);
String letter = list.stream().findAny().orElse(null);
System.out.println(letter);

- adrianosymphony

2

如果你必须使用流（streams），我写了一个优雅但非常低效的收集器来完成这项任务：

/**
 * Returns a random item from the stream (or null in case of an empty stream).
 * This operation can't be lazy and is inefficient, and therefore shouldn't
 * be used on streams with a large number or items or in performance critical sections.
 * @return a random item from the stream or null if the stream is empty.
 */
public static <T> Collector<T, List<T>, T> randomItem() {
    final Random RANDOM = new Random();
    return Collector.of(() -> (List<T>) new ArrayList<T>(), 
                              (acc, elem) -> acc.add(elem),
                              (list1, list2) -> ListUtils.union(list1, list2), // Using a 3rd party for list union, could be done "purely"
                              list -> list.isEmpty() ? null : list.get(RANDOM.nextInt(list.size())));
}

使用方法：

@Test
public void standardRandomTest() {
    assertThat(Stream.of(1, 2, 3, 4).collect(randomItem())).isBetween(1, 4);
}

- KidCrippler

1

有时候你可能想要在流中获取一个随机的项。如果你想要在过滤列表后仍然获取随机的项，那么这段代码片段将适用于你：

List<String> items = Arrays.asList("A", "B", "C", "D", "E");
List<String> shuffledAndFilteredItems = items.stream()
        .filter(value -> value.equals("A") || value.equals("B"))
        //filter, map...
        .collect(Collectors.collectingAndThen(
                Collectors.toCollection(ArrayList::new),
                list -> {
                    Collections.shuffle(list);
                    return list;
                }));
String randomItem = shuffledAndFilteredItems
        .stream()
        .findFirst()
        .orElse(null);

当然可能有更快/优化的方法，但它允许您一次完成所有操作。

- Atakan Yildirim

1

如果您事先不知道列表的大小，可以这样做：

 yourStream.collect(new RandomListCollector<>(randomSetSize));

我猜你需要编写自己的收集器实现，就像这个例子一样，以获得均匀随机化：

 public class RandomListCollector<T> implements Collector<T, RandomListCollector.ListAccumulator<T>, List<T>> {

private final Random rand;
private final int size;

public RandomListCollector(Random random , int size) {
    super();
    this.rand = random;
    this.size = size;
}

public RandomListCollector(int size) {
    this(new Random(System.nanoTime()), size);
}

@Override
public Supplier<ListAccumulator<T>> supplier() {
    return () -> new ListAccumulator<T>();
}

@Override
public BiConsumer<ListAccumulator<T>, T> accumulator() {
    return (l, t) -> {
        if (l.size() < size) {
            l.add(t);
        } else if (rand.nextDouble() <= ((double) size) / (l.gSize() + 1)) {
            l.add(t);
            l.remove(rand.nextInt(size));
        } else {
            // in any case gSize needs to be incremented
            l.gSizeInc();
        }
    };

}

@Override
public BinaryOperator<ListAccumulator<T>> combiner() {
    return (l1, l2) -> {
        int lgSize = l1.gSize() + l2.gSize();
        ListAccumulator<T> l = new ListAccumulator<>();
        if (l1.size() + l2.size()<size) {
            l.addAll(l1);
            l.addAll(l2);
        } else {
            while (l.size() < size) {
                if (l1.size()==0 || l2.size()>0 && rand.nextDouble() < (double) l2.gSize() / (l1.gSize() + l2.gSize())) {
                    l.add(l2.remove(rand.nextInt(l2.size()), true));
                } else {
                    l.add(l1.remove(rand.nextInt(l1.size()), true));
                }
            }
        }
        // set the gSize of l :
        l.gSize(lgSize);
        return l;

    };
}

@Override
public Function<ListAccumulator<T>, List<T>> finisher() {

    return (la) -> la.list;
}

@Override
public Set<Characteristics> characteristics() {
    return Collections.singleton(Characteristics.CONCURRENT);
}

static class ListAccumulator<T> implements Iterable<T> {
    List<T> list;
    volatile int gSize;

    public ListAccumulator() {
        list = new ArrayList<>();
        gSize = 0;
    }

    public void addAll(ListAccumulator<T> l) {
        list.addAll(l.list);
        gSize += l.gSize;

    }

    public T remove(int index) {
        return remove(index, false);
    }

    public T remove(int index, boolean global) {
        T t = list.remove(index);
        if (t != null && global)
            gSize--;
        return t;
    }

    public void add(T t) {
        list.add(t);
        gSize++;

    }

    public int gSize() {
        return gSize;
    }

    public void gSize(int gSize) {
        this.gSize = gSize;

    }

    public void gSizeInc() {
        gSize++;
    }

    public int size() {
        return list.size();
    }

    @Override
    public Iterator<T> iterator() {
        return list.iterator();
    }
}

}

如果您想要更简单的方法，但仍不想在内存中加载所有列表：

public <T> Stream<T> getRandomStreamSubset(Stream<T> stream, int subsetSize) {
    int cnt = 0;

    Random r = new Random(System.nanoTime());
    Object[] tArr = new Object[subsetSize];
    Iterator<T> iter = stream.iterator();
    while (iter.hasNext() && cnt <subsetSize) {
        tArr[cnt++] = iter.next();          
    }

    while (iter.hasNext()) {
        cnt++;
        T t = iter.next();
        if (r.nextDouble() <= (double) subsetSize / cnt) {
            tArr[r.nextInt(subsetSize)] = t;                

        }

    }

    return Arrays.stream(tArr).map(o -> (T)o );
}

但是这样你就离开了流API，并且可以使用基本迭代器完成相同的操作。

- Chris A

1

另一个想法是实现自己的 Spliterator，然后将其用作 Stream 的源：

import java.util.List;
import java.util.Random;
import java.util.Spliterator;
import java.util.function.Consumer;
import java.util.function.Supplier;

public class ImprovedRandomSpliterator<T> implements Spliterator<T> {

    private final Random random;
    private final T[] source;
    private int size;

    ImprovedRandomSpliterator(List<T> source, Supplier<? extends Random> random) {
        if (source.isEmpty()) {
            throw new IllegalArgumentException("RandomSpliterator can't be initialized with an empty collection");
        }
        this.source = (T[]) source.toArray();
        this.random = random.get();
        this.size = this.source.length;
    }

    @Override
    public boolean tryAdvance(Consumer<? super T> action) {
        if (size > 0) {
            int nextIdx = random.nextInt(size); 
            int lastIdx = size - 1; 

            action.accept(source[nextIdx]); 
            source[nextIdx] = source[lastIdx]; 
            source[lastIdx] = null; // let object be GCed 
            size--;
            return true;
        } else {
            return false;
        }
    }

    @Override
    public Spliterator<T> trySplit() {
        return null;
    }

    @Override
    public long estimateSize() {
        return source.length;
    }

    @Override
    public int characteristics() {
        return SIZED;
    }
}


public static <T> Collector<T, ?, Stream<T>> toShuffledStream() {
    return Collectors.collectingAndThen(
      toCollection(ArrayList::new),
      list -> !list.isEmpty()
        ? StreamSupport.stream(new ImprovedRandomSpliterator<>(list, Random::new), false)
        : Stream.empty());
}

然后简单地：

list.stream()
  .collect(toShuffledStream())
  .findAny();

详细信息可以在这里找到。

...但这肯定是过度的，所以如果你正在寻找实用的方法。一定要选择Jean的解决方案。

- Grzegorz Piwowarek

0

所选答案在其流解决方案中存在错误... 您不能使用Random#nextInt与非正长整型，即在此情况下为“0”。流解决方案也永远不会选择列表中的最后一个例子：

List<Integer> intList = Arrays.asList(0, 1, 2, 3, 4);

// #nextInt is exclusive, so here it means a returned value of 0-3
// if you have a list of size = 1, #next Int will throw an IllegalArgumentException (bound must be positive)
int skipIndex = new Random().nextInt(intList.size()-1);

// randomInt will only ever be 0, 1, 2, or 3. Never 4
int randomInt = intList.stream()
                       .skip(skipIndex) // max skip of list#size - 2
                       .findFirst()
                       .get();

我的建议是采用Jean-Baptiste Yunès提出的非流式方法，但如果你必须采用流式方法，你可以像这样做（但它有点丑陋）：

list.stream()
    .skip(list.isEmpty ? 0 : new Random().nextInt(list.size()))
    .findFirst();

- Tarandrus

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Jean-Baptiste Yunès · Accepted Answer

为什么要使用流？你只需要从0到列表的大小中获取一个随机数，然后调用此索引上的get函数：

Random r = new Random();
ElementType e = list.get(r.nextInt(list.size()));

流在这里不会给你带来任何有趣的东西，但你可以尝试使用：

Random r = new Random();
ElementType e = list.stream().skip(r.nextInt(list.size())).findFirst().get();

思路是跳过任意数量的元素（但不是最后一个元素！），然后获取第一个元素（如果存在）。结果将是Optional<ElementType>，该值将不为空，然后使用get提取其值。在跳过后，您有很多选择。

在这里使用流非常低效...

注意：这些解决方案都没有考虑空列表，但问题是针对非空列表定义的。