使用 pandas-gbq 執行查詢

使用 pandas-gbq 套件執行簡單查詢。

程式碼範例

Python

在試用這個範例之前，請先按照「使用用戶端程式庫的 BigQuery 快速入門導覽課程」中的 Python 設定說明操作。詳情請參閱 BigQuery Python API 參考說明文件。

如要向 BigQuery 進行驗證，請設定應用程式預設憑證。詳情請參閱「設定用戶端程式庫的驗證作業」。

import bigframes.pandas as bpd
import pandas as pd
import pandas_gbq

# Set partial ordering mode for BigQuery DataFrames.
bpd.options.bigquery.ordering_mode = "partial"


def read_gbq_simple_pandas_gbq() -> pd.DataFrame:
    """Queries BigQuery using pandas-gbq and returns an in-memory DataFrame."""
    sql = """
    SELECT name, SUM(number) as total_people
    FROM `bigquery-public-data.usa_names.usa_1910_2013`
    WHERE state = 'TX'
    GROUP BY name
    ORDER BY total_people DESC
    LIMIT 20
    """
    df_pandas = pandas_gbq.read_gbq(sql)
    print("Queried using pandas-gbq.")
    return df_pandas


def read_gbq_simple_bigframes() -> bpd.DataFrame:
    """Queries BigQuery using BigQuery DataFrames and returns a distributed DataFrame."""
    sql = """
    SELECT name, SUM(number) as total_people
    FROM `bigquery-public-data.usa_names.usa_1910_2013`
    WHERE state = 'TX'
    GROUP BY name
    ORDER BY total_people DESC
    LIMIT 20
    """
    df_bigframes = bpd.read_gbq(sql)
    print("Queried using BigQuery DataFrames.")
    return df_bigframes


# [Preferred] Run using pandas-gbq:
# df_pandas = read_gbq_simple_pandas_gbq()
# print(df_pandas.head())

# Alternatively, run using BigQuery DataFrames:
# df_bigframes = read_gbq_simple_bigframes()
# print(df_bigframes.head())

後續步驟

如要搜尋及篩選其他 Google Cloud 產品的程式碼範例，請參閱Google Cloud 範例瀏覽工具。