minerva/test.py
2023-07-27 13:46:44 -04:00

15 lines
527 B
Python

import access as a
import pprint
pp = pprint.PrettyPrinter(indent=4)
athena = a.Athena("hay", "s3://haystac-pmo-athena/")
query = athena.query('select * from "trajectories"."kitware" limit 10')
data = query.results()
print(data.head(10))
# Everything *needs* to have a column in order for parquet to work, so scalar
# values have to be assigned something, so here we use `as count` to create
# a temporary column called `count`
print(athena.query("select count(*) as count from trajectories.kitware").results().head(1))