从DataLab上传Numpy矩阵到BigQuery

!pip install numpy
!pip install google-cloud-bigquery

import numpy as np
from google.cloud import bigquery

matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

client = bigquery.Client()

table_id = 'your-project.your-dataset.your-table'  # 替换为你的项目、数据集和表格名称

schema = [
    bigquery.SchemaField('column1', 'INTEGER'),
    bigquery.SchemaField('column2', 'INTEGER'),
    bigquery.SchemaField('column3', 'INTEGER')
]

table = bigquery.Table(table_id, schema=schema)

rows = []

for row in matrix:
    rows.append({'column1': row[0], 'column2': row[1], 'column3': row[2]})

table.insert_data(rows)

job = client.insert_rows(table, rows)
job.result()  # 等待插入完成