Skip to content

Instantly share code, notes, and snippets.

@1ambda

1ambda/df-array-element_at.py

Created December 21, 2021 23:29
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save 1ambda/e26a9320e7fae58985f158c1cb2644e1 to your computer and use it in GitHub Desktop.
Save 1ambda/e26a9320e7fae58985f158c1cb2644e1 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
df-array-element_at.py
dfRaw\
    .withColumn("product_id", explode(col("product_id_set")))\
    .withColumn("category_element", explode(col("category_set")))\
    .withColumn("category_code", element_at(col("category_element"), 1))\
    .withColumn("category_id", element_at(col("category_element"), 2))\
    .select("product_id", "category_code", "category_id")\
    .show(truncate=False)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment