Skip to content

Instantly share code, notes, and snippets.

@fa-ahmad

fa-ahmad/ipython-sql.md

Last active February 11, 2022 10:25
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save fa-ahmad/67ea3b79c5684c8c6665eeaa1bc83eb0 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save fa-ahmad/67ea3b79c5684c8c6665eeaa1bc83eb0 to your computer and use it in GitHub Desktop.
Download ZIP
Capture output of SQL command with Ipython/Jupyter Notebook
Raw
ipython-sql.md

The indirection operator captures the output of any SQL query into a Python variable. In the example below the variable output will contain the result of the query. The query used here is irrelevant and only used for illustrating the usage.

%%sql output << 
SELECT a.attname, format_type(a.atttypid, a.atttypmod) AS data_type, a.attnotnull, i.indisprimary
FROM   pg_index i
JOIN   pg_attribute a ON a.attrelid = i.indrelid
                     AND a.attnum = ANY(i.indkey)
WHERE  i.indrelid = 'users'::regclass

Source

The variable output has the type sql.run.ResultSet and has builtin methods to convert it into a Pandas Dataframe or dump it as a CSV

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment