如何为 apache 光束数据流的输出 csv 添加标头？

Question

我注意到在 java sdk 中，有一个函数可以让您编写 csv 文件的标题。 https://cloud.google.com/dataflow/java-sdk/JavaDoc/com/google/cloud/dataflow/sdk/io/TextIO.Write.html#withHeader-java.lang.String-

此功能是否反映在 python skd 上？

Answer 1

您现在可以写入文本并使用文本接收器指定标题。

从文档：

class apache_beam.io.textio.WriteToText(file_path_prefix, file_name_suffix='', append_trailing_newlines=True, num_shards=0, shard_name_template=None, coder=ToStringCoder, compression_type='auto', header=None)

因此，您可以使用以下代码：

beam.io.WriteToText(bucket_name, file_name_suffix='.csv', header='colname1, colname2')

如果您需要详细信息或检查它的实现方式，可以在此处获得完整的文档：https : //beam.apache.org/documentation/sdks/pydoc/2.0.0/_modules/apache_beam/io/textio.html#WriteToText