SQL 表到 Pandas DataFrame

Question

我正在使用帶有 Psycopg2 庫的 PostgreSQL 數據庫。 我想獲取 SQL 表並將其轉換為 pd.DataFrame。 所以我需要將 SQL 表轉換為像{col1:[val1,val2], col2:[val1,val2]}這樣的字典，但是當我使用SELECT * FROM tablename時，它不包含列名。 我不想使用 pandas 函數直接連接到 SQL。 如果您知道如何獲得 SQL 表作為我上面提到的字典，如果您寫的話，我很高興。

提前感謝您的幫助。

Answer 1

執行查詢后，您可以從description -attribute 中的 cursor 中提取列名：

cursor.execute("SELECT * FROM tablename")
columns = [desc[0] for desc in cursor.description]

之后，您應該能夠使用columns參數將獲取的數據解析為具有正確列名的 DataFrame：

data_df = pd.DataFrame(cursor.fetchall(), columns=columns)

最后使用to_dict將 DataFrame 轉換為以列表為值的字典：

data_dict = data_df.to_dict(orient="list")