pandas_gbq_quickstart_dataframe_to_bigquery_roundtrip.py

1import pandas as pd
2import pandas_gbq
3
4# Create a sample DataFrame
5data = {
6    "my_string": ["a", "b", "c"],
7    "my_int64": [1, 2, 3],
8    "my_float64": [4.0, 5.0, 6.0],
9    "my_bool": [True, False, True],
10}
11df = pd.DataFrame(data)
12
13# TODO: Set project_id to your Google Cloud Platform project ID.
14project_id = "your-project-id"
15
16# 1. Write the DataFrame to a BigQuery table
17# This will create the table if it doesn't exist.
18table_id = "my_dataset.my_table"
19pandas_gbq.to_gbq(df, table_id, project_id=project_id)
20
21# 2. Read data from BigQuery back into a DataFrame
22sql = f"SELECT * FROM `{table_id}`"
23df_from_bq = pandas_gbq.read_gbq(sql, project_id=project_id)
24
25print(df_from_bq)