如何使用gsutil写一个查询到表?

时间:2018-01-29 19:53:03

标签: python pandas google-bigquery gsutil

我目前正在使用python和pandas构建查询,将结果转换为pandas dataframe,然后将dataframe写回表。要跳过下载和上传步骤,我想我应该使用gsutil执行查询并指示它将结果写入表。

我正在使用的查询是:

SELECT id, weight, net_weight
FROM `body_table-1345.bodies.weights
WHERE birthdate >= '2017-01-01 00:00:00'

我想将结果写入名为body_table-1345.bodies.eligible的表中。

我当前的代码使用的是pandas.DataFrame.read_gbqpandas.DataFrame.to_gbq,但这需要太长时间,我认为这样做是错误的。一旦我在命令行中执行了正确的操作(使用gsutil),我将把它包装在python代码中并使用os.system执行它。

如何告诉gsutil自动检测SCHEMA并将结果发送到表格?

1 个答案:

答案 0 :(得分:2)

我建议使用官方Google Cloud Python API,而不是使用Python作为os命令的包装器。您可以非常轻松地执行所有这些步骤,例如:

import google.cloud.bigquery as bq
client = bq.Client.from_service_account_json('path/to/credentials.json')

table = client.dataset("bodies").table("eligible")

query_config = bq.QueryJobConfig()
query_config.destination = table
query = """
    SELECT id, weight, net_weight
    FROM `body_table-1345.bodies.weights
    WHERE birthdate >= '2017-01-01 00:00:00'
    """
job = client.query(query, job_config=query_config)

您可以在official docs其他几个示例中找到如何使用API​​执行这些操作。