snowpark_session_setup_with_dataframe_filter_and_select.py

1from snowflake.snowpark import Session
2
3# Replace the connection_parameters with your actual Snowflake account details
4connection_parameters = {
5    "account": "<your_account_identifier>",
6    "user": "<your_username>",
7    "password": "<your_password>",
8    "role": "<your_role>",
9    "warehouse": "<your_warehouse>",
10    "database": "<your_database>",
11    "schema": "<your_schema>"
12}
13
14# Create a Snowpark Session
15session = Session.builder.configs(connection_parameters).create()
16
17# Create a sample DataFrame
18df = session.create_dataframe([[1, "Name 1"], [2, "Name 2"]], schema=["ID", "NAME"])
19
20# Perform a basic transformation (Filter and Select)
21df_filtered = df.filter(df["ID"] == 1).select("NAME")
22
23# Show the results
24df_filtered.show()
25
26# Close the session
27session.close()