在Apache Beam Python SDK中，没有与withFormatFunction函数等效的函数。然而，你可以使用ParDo转换和自定义函数来达到类似的效果。

下面是一个使用ParDo和自定义函数的示例代码：

import apache_beam as beam

class MyFormatFunction(beam.DoFn):
    def process(self, element):
        # 在这里编写自定义的格式化逻辑
        # 返回格式化后的结果
        return [formatted_element]

def main():
    with beam.Pipeline() as p:
        data = p | beam.Create(['element1', 'element2'])

        formatted_data = data | beam.ParDo(MyFormatFunction())

        formatted_data | beam.io.WriteToText('output.txt')

if __name__ == '__main__':
    main()

在上面的代码中，MyFormatFunction是自定义的格式化函数，它继承自beam.DoFn类，并重写了process方法。在process方法中，你可以编写自己的格式化逻辑，并返回格式化后的结果。

然后，通过将data输入到ParDo转换中，并传入MyFormatFunction，可以实现对数据的格式化处理。最后，将格式化后的数据输出到文本文件中。

请注意，这只是一个示例，你可以根据实际需求编写自定义的格式化函数。