python - AWS Glue,输出一个带分区的文件
问题描述
我有一个 Glue ETL 脚本,它采用分区的 Athena 表并将其输出到 CSV。该表按两个标准进行分区,单元和站点。当 Glue 作业运行时,它会为单元和站点分区的每个组合创建一个不同的 CSV 文件。相反,我只想要一个包含所有分区的输出文件,类似于 athena 表的结构
我已经对“datasource0.toDF().repartition(1)”进行了一些修改,但我不确定它如何与 AWS 提供的脚本交互。我已经用镶木地板做了这个,但是这个脚本的结构不同
注意下面的脚本我已经删除了大部分标签映射
from awsglue.transforms import *
from awsglue.utils import getResolvedOptions
from pyspark.context import SparkContext
from awsglue.context import GlueContext
from awsglue.job import Job
## @params: [JOB_NAME]
args = getResolvedOptions(sys.argv, ['JOB_NAME'])
sc = SparkContext()
glueContext = GlueContext(sc)
spark = glueContext.spark_session
job = Job(glueContext)
job.init(args['JOB_NAME'], args)
## @type: DataSource
## @args: [database = "testdata-2018-2019", table_name = "testdata", transformation_ctx = "datasource0"]
## @return: datasource0
## @inputs: []
datasource0 = glueContext.create_dynamic_frame.from_catalog(database = "formatted-test-2018-2019", table_name = "testdata", transformation_ctx = "datasource0")
datasource0.toDF().repartition(1)
## @type: ApplyMapping
## @args: [mapping = [("time", "string", "time", "string"), ("unit", "string", "unit", "string")], transformation_ctx = "applymapping1"]
## @return: applymapping1
## @inputs: [frame = datasource0]
applymapping1 = ApplyMapping.apply(frame = datasource0, mappings = [("time", "string", "time", "string"), ("`data.pv`", "double", ("site", "string", "site", "string"), ("unit", "string", "unit", "string")], transformation_ctx = "applymapping1")
## @type: DataSink
## @args: [connection_type = "s3", connection_options = {"path": "s3://testbucket/ParsedCSV-Data"}, format = "csv", transformation_ctx = "datasink2"]
## @return: datasink2
## @inputs: [frame = applymapping1]
datasink2 = glueContext.write_dynamic_frame.from_options(frame = applymapping1, connection_type = "s3", connection_options = {"path": "s3://buckettest/ParsedCSV-Data"}, format = "csv", transformation_ctx = "datasink2").repartition(1)
job.commit()
我想修改上述脚本以仅输出一个包含分区列的 CSV 文件。我怎样才能做到这一点?
解决方案
您需要在编写 DynamicFrame之前重新分区。
repartitioned1 = applymapping1.repartition(1)
datasink2 = glueContext.write_dynamic_frame.from_options(frame = repartitioned1, connection_type = "s3", connection_options = {"path": "s3://20182019testdata/ParsedCSV-Data"}, format = "csv", transformation_ctx = "datasink2")
关于将分区列包含到输出文件中,我认为这是不可能的。作为一种解决方法,您可以将一列复制到具有不同名称的新列中。
df = applymapping1.toDF
repartitioned_with_new_column_df = df.withColumn("_column1", df["column1"]).repartition(1)
dyf = DynamicFrame.fromDF(repartitioned_with_new_column_df, glueContext, "enriched")
datasink2 = glueContext.write_dynamic_frame.from_options(frame = dyf, connection_type = "s3", connection_options = {"path": "s3://20182019testdata/ParsedCSV-Data", , "partitionKeys": ["_column1"]}, format = "csv", transformation_ctx = "datasink2")
推荐阅读
- c# - 在 c# .net core 中创建自定义文件扩展名
- tmlanguage - 有没有办法防止(除了).tmLanguage 中“匹配”中的特定字符
- html - 将文本添加到边框而不添加间距
- php - 接收推送通知两次 firebase PHP 脚本
- jsp - 将 Web 应用程序与 IdentityServer 集成
- elasticsearch - 未找到“弹性搜索”模块
- ios - 如何修复——当权限处理程序已经在我的 Podfile 中时,IPHONEOS_DEPLOYMENT_TARGET 设置为 8.0?
- reactjs - 在 React Js 中更改 Segment 的图标
- c# - EF Core LINQ GROUPBY 然后选择获取实体的多个属性
- python - 将数组字典转换为单个熊猫数据框