sql：选择列列表

Question

我想传递一个 str 或 list 参数，并希望 sql 知道如何处理它。 list_col='date1, date2, date3, date4' 的示例，最后我想要数据框 日期1，日期2，日期3，身份证

query = """
    SELECT {list_col} AT TIME ZONE 'Europe/Paris' as {list_col}, {table}.{id}
    FROM {table} 
    ORDER BY {table}.{id}
"""

def fun_query(table_name, list_col, id):
    return query.format(table=table_name, list_col=list_col, id=id)

请问有人知道怎么做吗？

Answer 1

正如已经指出的那样，这在您建议的方式中是不可行的，因为AT TIME ZONE 和AS 子句都应与每列一起出现。我建议做这样的事情。

query = """
    SELECT {date_cols_as_tz}, {table}.{id}
    FROM {table} 
    ORDER BY {table}.{id}
"""


def fun_query(table_name, list_col, id, tz="'Europe/Paris'"):
    date_cols_as_tz = ",".join((f"{c} AT TIME ZONE {tz} as {c}" for c in list_col))
    return query.format(date_cols_as_tz=date_cols_as_tz, table=table_name, list_col=list_col, id=id)

当你打电话时，例如fun_query("my_table", ["date1", "date2"], "table_id") 并打印出来你会得到以下查询：

SELECT date1 AT TIME ZONE 'Europe/Paris' as date1,date2 AT TIME ZONE 'Europe/Paris' as date2, my_table.table_id
FROM my_table 
ORDER BY my_table.table_id

主要变化是：

• 在fun_query 内创建date_cols_as_tz
• 对list_col 参数使用真实列表（不是"date1,date2" 之类的字符串，而是["date1", "date2"] 之类的列表）
• 为函数添加了可选的tz 参数

此解决方案的优点是您可以通过为tz 使用不同的值而不是硬编码值来轻松更改时区。

另请注意，此函数要求 list_col 中的所有列都是日期（但如果我正确理解您的问题，这可能是您所期望的）。