正如@Andrew在他的评论中所解释的那样，也许你可以在Apache Beam中使用Top转换（对于Java或Python）。具体来说，Top.of()函数返回一个 * PTransom * 和一个 PCollection，按比较器转换排序。
在这里你可以找到一个简单的使用示例：

PCollection<Student> students = ...;
PCollection<List<Student>> top10Students = students.apply(Top.of(10, new CompareStudentsByAvgGrade()));

here是另一个使用Apache Beam Python SDK的例子，它解决了 PCollection 中返回单个元素的问题。

对于X元素的随机样本，可以使用内置的Sample转换（对于Python或Java）。
下面是一个示例，展示了如何从100个元素的示例输入中采样10个元素：

import apache_beam as beam
from apache_beam.transforms.combiners import Sample

with beam.Pipeline(runner='DirectRunner') as p:
    input = p | beam.Create(range(100))
    output = input | Sample.FixedSizeGlobally(10)
    output | beam.io.WriteToText('output')

如果你不关心顺序，只是想要N个项目的样本，那么我认为你应该能够使用beam.combiners.Sample.FixedSizeGlobally，如下所述：https://beam.apache.org/documentation/transforms/python/aggregation/sample/