diff --git a/examples/athena_basic_query.py b/examples/athena_basic_query.py index c48e9d2..11ca08b 100644 --- a/examples/athena_basic_query.py +++ b/examples/athena_basic_query.py @@ -13,17 +13,15 @@ athena = m.athena("s3://haystac-pmo-athena/") # Everything *needs* to have a column in order for unloading to parquet to work, # so scalar values have to be assigned something, so here we use `as count` to # create a temporary column called `count` -#query = athena.query( -# """ -# select round(longitude, 3) as lon, count(*) as count -# from trajectories.basline -# where agent = 4 -# group by round(longitude, 3) -# order by count(*) desc -# """ -#) - -query = athena.query("select * from trajectories.basline where agent < 100 limit 100") +query = athena.query( + """ + select round(longitude, 3) as lon, count(*) as count + from trajectories.basline + where agent = 4 + group by round(longitude, 3) + order by count(*) desc + """ +) data = query.results() pp.pprint(data.head(10))