将列动态传递给python postgresql中的select()

import psycopg2

def dynamic_select(columns, table_name):
    # 连接到PostgreSQL数据库
    conn = psycopg2.connect(
        dbname="your_database",
        user="your_user",
        password="your_password",
        host="your_host",
        port="your_port"
    )
    cursor = conn.cursor()

    # 构建动态的SELECT语句
    columns_str = ", ".join(columns)
    query = f"SELECT {columns_str} FROM {table_name}"

    # 执行查询
    cursor.execute(query)

    # 获取结果
    results = cursor.fetchall()

    # 关闭连接
    cursor.close()
    conn.close()

    return results

# 示例调用
columns_to_select = ["id", "name", "age"]
table_name = "users"
results = dynamic_select(columns_to_select, table_name)
print(results)