我们可以使用以下代码,在BigQuery中返回针对特定客户列表的记录。
首先,创建一个包含客户ID的列表,比如下面这样:
customers_to_retrieve = ['12345', '67890', '54321']
然后,使用以下查询获取数据:
SELECT *
FROM mydataset.mytable
WHERE customer_id IN UNNEST(@customers_to_retrieve)
将客户列表作为参数传递给查询:
query_params = [ bigquery.ArrayQueryParameter('customers_to_retrieve', 'STRING', customers_to_retrieve) ]
job_config = bigquery.QueryJobConfig() job_config.query_parameters = query_params
执行查询,获取结果:
query_job = client.query(query, location='US', job_config=job_config)
results = query_job.result()
最后,可以对结果进行处理,例如将其保存为CSV文件:
with open('results.csv', 'w') as f: for row in results: f.write(','.join([str(x) for x in row.values()]) + '\n')
这样,我们就可以方便地在BigQuery中获取针对特定客户列表的记录。