import boto3
import pandas as pd
from pyathena import connect
from pyathena.pandas.cursor import PandasCursor

# AWS セッションの設定 (必要に応じてプロファイル名を指定)
session = boto3.Session(profile_name="your-profile", region_name="ap-northeast-1")

# Athena 接続の設定
conn = connect(
    aws_access_key_id=session.get_credentials().access_key,
    aws_secret_access_key=session.get_credentials().secret_key,
    aws_session_token=session.get_credentials().token,
    region_name="ap-northeast-1",
    s3_staging_dir="s3://your-bucket/",
    schema_name="default",
    cursor_class=PandasCursor
)

# クエリの実行
query = """
select 
    *
from
    "your-database"."your-table"
where  
    timestamp >= from_iso8601_timestamp('2024-01-01T00:00:00Z') and
    user is not null
"""

# Pandas DataFrame にクエリ結果をロード
df = pd.read_sql(query, conn)