如何将 postgresql 数据库的列放入 Python 列表中？

Question

我正在尝试从 PostgreSQL 数据库中获取表的列名，然后将它们放入 python 列表中。 我真的不知道如何获取列名。 我试过了：

select * from information_schema.columns where table_name = "my_table"

我认为可能有更简单的方法。 你知道如何轻松做到吗？

我遇到的下一个问题是我想在 Python 中完成所有这些工作，而且我对它有点陌生。 我的目标是将列名保存在列表中。

先感谢您。

Answer 1

您的 SQL 是正确的，这是最方便的方法。

执行查询后，只需 append 将它们添加到某个列表

columns = []
rows = db.execute('select * from information_schema.columns where table_name = "my_table"')
for row in rows:
    columns.append(row['column_name']) # or row[0] i think.

print(columns)