从 python 函数中返回 DataFrame 而不是使用 Pandas 的 pd.read_sql_query 列表

Question

我很难理解为什么这会返回一个包含 DataFrame 而不仅仅是 DataFrame 的列表。 代码或仅返回 DataFrame 的方法有问题吗？ 如果未放置在函数中，它会按预期工作。

import sqlite3
import pandas as pd

def get_tbl_info(db ='MyDatabase', table ='Measurements'):

        database = "/Users/Mary/Documents/Database/{DB}.db"..format(DB=db)

        conn = sqlite3.connect(database)

        tbl_info_command = "PRAGMA TABLE_INFO({table});".format(table=table)
        result_all = pd.read_sql_query(tbl_info_command,conn)
        print(type(result_all))
        return [result_all]



out = get_tbl_info()
print(type(out))

给出：

<class 'pandas.core.frame.DataFrame'>
<class 'list'>

Answer 1

因为当您用方括号[ ]括起一个变量时，Python 会将其理解为“可以将此变量放入列表中” 。 只需将函数的return替换为：

return result_all

这应该可以正常工作，或者实际上您的函数可以直接返回您的数据帧

return pd.read_sql_query(tbl_info_command,conn)