使用Apache Beam，您可以使用GroupByKey将键值对按键分组，然后使用ParDo将每个组的值写入文件。

以下是一个示例代码：

import apache_beam as beam

def write_to_file(element):
    key, values = element
    with open(f'{key}.txt', 'w') as f:
        for value in values:
            f.write(f'{value}\n')

with beam.Pipeline() as pipeline:
    data = pipeline | beam.Create([
        ('key1', 'value1'),
        ('key2', 'value2'),
        ('key1', 'value3'),
        ('key2', 'value4')
    ])
    
    results = data | beam.GroupByKey() | beam.ParDo(write_to_file)

在这个示例中，我们首先创建一个PCollection，其中包含键值对。然后，我们使用GroupByKey将键值对按键分组。最后，我们使用ParDo将每个组的值写入文件。在write_to_file函数中，我们使用键作为文件名，并将每个值写入该文件。

这将生成两个文件：key1.txt和key2.txt。key1.txt文件将包含值value1和value3，而key2.txt文件将包含值value2和value4。

注意：在实际使用中，您可能需要根据您的需求修改write_to_file函数来适应更复杂的写入逻辑。此示例仅演示了一个简单的写入操作。