Running AWS Athena SQL queries

athena_sagamaker.py

import sys
!{sys.executable} -m pip install PyAthena

from pyathena import connect
import pandas as pd
conn = connect(s3_staging_dir='s3://dev-sample-data123/temp',
               region_name='ap-south-1')

df = pd.read_sql("SELECT * FROM rank_table limit 10;", conn)

# skipping header line
df1 = df.iloc[1:]
# changing datatype of rank to int
df1['rank'] = df1['rank'].astype(int)

df1.plot()