如何在python中比较同一个PCollection中两个key的所有值？

import apache_beam as beam

def compare_values(element):
    key1, key2 = element
    values1, values2 = key1[1], key2[1]
    
    # 比较两个key的所有值
    if sorted(values1) == sorted(values2):
        return (key1, key2)
    else:
        return None

with beam.Pipeline() as pipeline:
    # 创建一个PCollection，包含两个key和对应的值
    input_data = [
        (('key1', [1, 2, 3]), ('key2', [3, 2, 1])),
        (('key3', [4, 5, 6]), ('key4', [6, 5, 4])),
        # 更多的数据...
    ]
    pcollection = pipeline | beam.Create(input_data)
    
    # 使用ParDo操作来比较两个key的所有值
    compared_values = pcollection | beam.ParDo(compare_values)
    
    # 输出比较结果
    compared_values | beam.Map(print)