>>> from snowflake.snowpark.window import Window
>>> df = session.create_dataframe(
... [
... [1, 2, 1],
... [1, 2, 3],
... [2, 1, 10],
... [2, 2, 1],
... [2, 2, 3],
... ],
... schema=["x", "y", "z"]
... )
>>> df.select(row_number().over(Window.partition_by(col("X")).order_by(col("Y"))).alias("result")).show()
------------
|"RESULT" |
------------
|1 |
|2 |
|3 |
|1 |
|2 |
------------