获取分区指定分区的数据

导入必要的库

import apache_beam as beam

创建一个管道

p = beam.Pipeline()

读取数据

data = p | beam.io.ReadFromText('data.txt')

将数据分区

partitioned_data = data | beam.Partition(lambda x: x[0], 10)

获取分区指定分区的数据

partitioned_data_1 = partitioned_data[1]

打印分区指定分区的数据

partitioned_data_1 | beam.io.WriteToText('partitioned_data_1.txt')

运行管道

p.run()