將 BigQuery 行轉換為以第一列為鍵的字典

Question

我正在使用 BigQuery Python API，並且我有一個返回 2 xn 行的 query_job。 保證第一列中的值是唯一的。 我想將結果轉換為 Python 字典，其中第一列中的值將成為字典鍵，第二列中的值將成為字典值。 當然，我可以在每一行上循環； 但我只是想知道是否有更優雅（閱讀：快捷方式）的解決方案。

Answer 1

您可以直接將從查詢返回的 object 轉換為dict ：

from google.cloud import bigquery

# Construct a BigQuery client object.
client = bigquery.Client()

query = """
    SELECT id, name
    FROM `project_id.dataset_id.table_id`
    ORDER BY id
    LIMIT 20
"""
query_job = client.query(query)  # Make an API request.

result_dict = dict(query_job)

這里假設id列具有將用作鍵的唯一值。

Answer 2

嘗試這個：

from google.cloud import bigquery

# Construct a BigQuery client object.
client = bigquery.Client()

query = """
        SELECT id, name
        FROM `project_id.dataset_id.table_id`
        ORDER BY id
        LIMIT 20
        """
query_job = client.query(query)  # Make an API request.

for row in query_job:
    print(dict(row.items()))