如何将包含多个ID号的Python变量传入外部BigQuery脚本查询

方案1：修改现有占位符替换逻辑（改动最小，可复用现有.sql脚本） #

核心逻辑是提前把ID（支持单个或多个）处理为符合SQL IN从句语法的字符串，替换后直接可用，不会产生多余符号：

1. 首先修改Python方法逻辑：

def state_data(self, state, id_number):
    # 统一处理ID输入：兼容单个字符串、列表、元组格式的入参
    if isinstance(id_number, (list, tuple)):
        # 多个ID场景：每个ID加单引号，逗号拼接
        formatted_ids = ",".join([f"'{id_val}'" for id_val in id_number])
    else:
        # 单个ID场景：直接加单引号
        formatted_ids = f"'{id_number}'"
    
    if state == 'NY': 
        sql_script = self.sql_scripts['get_data_ny']
    else:
        sql_script = self.sql_scripts['get_data_rest']
         
    sql_script = sql_script.replace('@@id_number@@', formatted_ids)

2. 调整.sql脚本中的WHERE从句，去掉多余的引号/括号，修改为：

WHERE ID_NUM IN(@@id_number@@)

替换后最终生成的SQL会自动符合语法要求，例：WHERE ID_NUM IN('123','124')

方案2：临时表关联方案（适合ID数量超过1000的场景，查询性能更优） #

可以复用你之前的临时表思路，调整占位符和处理逻辑即可：

1. 修改.sql脚本的INSERT部分，替换为VALUES占位符：

CREATE OR REPLACE TEMP TABLE ID_Numbers
(
     ID_Number STRING
);

INSERT INTO ID_Numbers (ID_Number)
VALUES @@values@@;

-- 后续关联逻辑不变

2. Python中处理ID生成对应的VALUES片段：

def state_data(self, state, id_number):
    # 处理ID生成VALUES片段
    if isinstance(id_number, (list, tuple)):
        values_str = ",".join([f"('{id_val}')" for id_val in id_number])
    else:
        values_str = f"('{id_number}')"
    
    if state == 'NY': 
        sql_script = self.sql_scripts['get_data_ny']
    else:
        sql_script = self.sql_scripts['get_data_rest']
         
    sql_script = sql_script.replace('@@values@@', values_str)

之前报错问题说明 #

你第三个方案的AttributeError: 'QueryJob' object has no attribute 'format'报错，是因为你调用format方法的对象是BigQuery返回的QueryJob类型，不是字符串，不是SQL语法的问题，调整变量引用顺序即可。

注意事项 #

如果ID本身包含单引号，需要提前做转义，避免SQL语法错误，将字符串格式化部分改为f"'{id_val.replace("'", "''")}'"即可。

内容的提问来源于stack exchange，提问作者coderX