Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save dvu4/a83c9bd817e6c8101c1c028b951cfbad to your computer and use it in GitHub Desktop.
Save dvu4/a83c9bd817e6c8101c1c028b951cfbad to your computer and use it in GitHub Desktop.
 window_spec = Window.partitionBy("rxa_claim_id", "ITEM_PRODUCT_CODE").orderBy(df_idi["EVENT_TIMESTAMP"].desc())
 df_valid_claim = df_valid_claim.withColumn("row_number", F.row_number().over(window_spec))
 df_valid_claim.display()

window_spec_2 = Window.partitionBy("rxa_claim_id", "ITEM_PRODUCT_CODE")
valid_claim =  df_valid_claim.withColumn("row_number_count", F.count("row_number").over(window_spec_2))
df_valid_claim.display()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment