Airflow Bigquery Hook:如何将结果保存在python变量中

Question

我在我的气流代码中使用bigquery钩子。

查询示例：select count(*) from 'table-name';因此它将只返回1个整数作为结果。

如何将其保存在Integer python变量中，而不是整个Panda数据帧中？

下面是我的代码示例

hook = BigQueryHook(bigquery_conn_id=BQ_CON, use_legacy_sql=False)
bq_client = bigquery.Client(project = hook._get_field("project"), credentials = hook._get_credentials())
query = "select count(*) from dataset1.table1;"
df = bq_client.query(query).to_dataframe()

Answer 1

如果它只是一行，您可以将列命名为col1，并使用此键名访问它

query = "select count(*) as col1 from dataset1.table1;"
query_result = client.query(query)
result = query_result[0]['col1']

或者如果您已经呼叫to_dataframe()

result = int(df.values[0])