Run SQL queries in an Athena notebook

Run saved SQL queries from a notebook to use their results in your analysis.

Step-by-step Instructions

Create a query snippet

Navigate to Spaces. Drag and drop the database you wish to query onto your Space. In the Chat sidebar, enter the following prompt and press Enter.

Prompt

Use query to answer: Which drivers have scored the most points so far in the 2024 season?

Database table in Spaces beside a SQL query request

Save the query

Save your query snippet by clicking on the save icon present on top-right of the query snippet.

SQL snippet editor with the save control highlighted

Get IDs

In the right Chat sidebar, enter the following prompt and press Enter.

Prompt

Retrieve the asset_id for @your-query-name and @your-database-name.

Replace your-query-name and your-database-name with the names of your query and database on which the query is executed.

Chat requesting the query and database asset IDs

Queries in notebooks can be executed either by referencing a saved SQL snippet or by directly running SQL statements against your connected databases.

Query a saved SQL snippet
Execute SQL against your DBs

Create a Notebook

In the right Chat sidebar, enter the following prompt and press Enter.

Prompt

Create a new Notebook.

New notebook open beside the saved SQL query

Execute your query from the saved query snippet

Copy and paste the following python snippet in a new cell of the Notebook that just got created. Run the cell.

from athena.client import Athena
# Setup the client
client = Athena ()
# Execute SQL snippet and load results into a DataFrame
df = client.query.execute_snippet(snippet_asset_id="query-asset-id")
#print the df
print(df)

Replace "query-asset-id" with the saved query’s asset ID.

Notebook cell executing a saved query snippet and displaying results

Create a Notebook

In the right Chat sidebar, enter the following prompt and press Enter.

Prompt

Create a new Notebook.

Format and execute your query

Use the database SQL execution endpoint to run SQL directly against your connected database. Refer to the Database SQL API documentation for details.

Learn

Core Concepts

Work in a Session

Connect Your Data

Analyze Data

Create and Edit

Meetings, Email and Voice

Build and Automate

Account and Security

Desktop and Mobile

Industry Solutions

Run SQL queries in an Athena notebook

Step-by-step Instructions

​Step-by-step Instructions

Step-by-step Instructions