psycopg2.errors.UndefinedColumn：列

Question

尝试获取所有数据时收到此错误消息

Traceback (most recent call last):
  File "/home/user/xxxxx/node_modules/serverless/lib/plugins/aws/invokeLocal/runtimeWrappers/invoke.py", line 86, in <module>
    result = handler(input['event'], context)
  File "./src/handler.py", line 97, in historical
    post_process(db_connection)
  File "./src/post_process.py", line 25, in post_process
    averages = db_connection.fetch_all("""
  File "./src/common/database_helper.py", line 36, in fetch_all
    cursor.execute(query)
psycopg2.errors.UndefinedColumn: column "col_list" does not exist
LINE 4:                                            avg(col_list[z])

代码

col_list = ['A', 'B', 'C']

averages = db_connection.fetch_all("""
                                           SELECT
                                           DATE_TRUNC('month', "DateTime") AS date_time,
                                           avg(col_list[z])
                                           FROM public."price_NA" group by DATE_TRUNC('month', "DateTime")
                                           ORDER BY date_time DESC
                                           """
                                           )

Answer 1

你真的认为这会奏效吗？ SQL 对 Python 变量一无所知。 幸运的是，使用 f-strings 很容易解决：

col_list = ['A', 'B', 'C']

averages = db_connection.fetch_all(f"""
                                           SELECT
                                           DATE_TRUNC('month', "DateTime") AS date_time,
                                           avg({col_list[z]})
                                           FROM public."price_NA" group by DATE_TRUNC('month', "DateTime")
                                           ORDER BY date_time DESC
                                           """
                                           )