【发布时间】:2021-01-01 20:19:15
【问题描述】:
目前使用python从bigquery中拉取数据,得到结果后需要转成JSON格式。如何将结果转成JSON?
query_job2 = client_bq.query(query)
query_job2.result()
rows = list(query_job2.result()) # Waits for query to finish
response = dict()
"""
Creating a nested dictionary with the tables as the keys and inside each respective table will hold cost as keys and
have a list of values
"""
for row in rows:
table = get_table_name(str(row.query))
start_time =int(row.start_time.timestamp())
end_time =int(row.end_time.timestamp())
if table in response:
if row.cost in response[table]:
response[table] = list(response[table])
response[table].append((str(row.creation_time),start_time,end_time , row.cost, str(row.query)))
else:
response[table] = {}
response[table] = (str(row.creation_time), start_time,end_time, row.cost, str(row.query))
这是我正在使用的查询:
with data AS (
SELECT
creation_time,
total_bytes_processed,
query
FROM `project.region-us.INFORMATION_SCHEMA.JOBS_BY_PROJECT`
where creation_time > TIMESTAMP_ADD(CURRENT_TIMESTAMP(), INTERVAL -60 SECOND) AND job_type = "QUERY"
Group BY creation_time, job_id, total_bytes_processed, query
ORDER BY total_bytes_processed DESC
)
select as value
array_agg(struct( creation_time,
regexp_extract(query, r'(?i)\sfrom\s+`?(?:[\w-]+\.)*([\w-]+\.[\w-]+)`?\s' ) as table,
(total_bytes_processed/1099511627776) * 5 as cost,
query) order by (total_bytes_processed/1099511627776) * 5 desc limit 1)[offset(0)]
from data
group by timestamp_trunc(creation_time, minute)
"""
【问题讨论】:
-
这个问题可以更好地表达。您在问如何将 pandas.DataFrame 格式化为 json。您可以通过调用 df.to_json(orient=....) 来实现。它与 BigQuery 无关,因为您可以等效地从 CSV 读取数据。
-
@gidutz 真的很有趣,我想也许我可以直接从查询或其他东西中获取结果,或者将 row.iterator 对象转换为字典或其他东西
标签: json python-3.x google-bigquery