pytd_treasure_data_presto_query_to_pandas_dataframe.py

1import pytd
2import pandas as pd
3
4# Initialize the client
5# Replace 'YOUR_API_KEY' and 'endpoint' with your actual Treasure Data credentials
6client = pytd.Client(apikey='YOUR_API_KEY', endpoint='https://api.treasuredata.com', database='sample_datasets')
7
8# Execute a query and retrieve results as a pandas DataFrame
9results = client.query('SELECT * FROM www_access LIMIT 10', engine='presto')
10
11# Display the data
12print(results['data'])
13print(results['columns'])
14
15# Alternatively, convert directly to a DataFrame if using the results object
16df = pd.DataFrame(**results)
17print(df.head())