Python新手求助：如何实现从SQL Server获取数据并转为DataFrame的可复用函数？

嘿，刚接触Python函数开发不用慌，我帮你梳理下代码里的问题，然后给你一个能正常跑起来的版本，完全满足后续其他函数调用DataFrame的需求~

先说说你现有代码里的几个关键问题：

• __init__方法缺少self参数，Python类的实例方法必须把self作为第一个参数，不然没法访问实例的属性和方法
• 连接字符串的写法错误，你只是把多行字符串分开写，但没拼接成完整的有效字符串，也没赋值给实例变量，后面根本调用不了
• select方法里重复定义了query变量，你传入的query参数直接被覆盖，等于白传了自定义查询的功能
• select里的con参数完全没用到，反而自己创建连接时，没通过self引用driver_path，会报变量未定义的错误
• 没有处理数据库连接的关闭，最好用上下文管理器自动管理连接，避免资源泄漏

下面是修正后的完整代码，我会一步步给你解释：

import pyodbc
import pandas as pd

class Inputdata:
    def __init__(self, server, database, trusted_connection="yes"):
        # 初始化时接收数据库连接参数，拼接完整的连接字符串
        self.connection_string = (
            f"Driver={{SQL Server}};"
            f"Server={server};"
            f"Database={database};"
            f"Trusted_Connection={trusted_connection};"
        )

    def select(self, query):
        # 使用上下文管理器（with语句）自动管理数据库连接，不用手动关闭
        with pyodbc.connect(self.connection_string) as con:
            # 用传入的自定义query执行查询，直接转成DataFrame
            df = pd.read_sql(query, con)
        return df

代码细节解释： #

1. 导入依赖库：必须先导入pyodbc（负责和SQL Server建立连接）和pandas（负责把查询结果转成DataFrame）
2. __init__方法优化：
   • 把服务器名、数据库名等作为参数传入，这样这个类可以灵活连接不同的数据库，不用硬写死配置
   • 用f-string拼接完整的连接字符串，作为实例的connection_string属性，后续方法可以通过self直接访问
3. select方法改进：
   • 只需要传入你要执行的SQL查询语句query，不用额外传连接参数
   • 用with pyodbc.connect(...)创建连接，上下文管理器会在代码块结束后自动关闭连接，避免资源浪费
   • 调用pd.read_sql把查询结果直接转成DataFrame，返回出去供其他函数调用

如何使用这个类： #

# 实例化Inputdata类，传入你的实际数据库信息
data_fetcher = Inputdata(server="xxx", database="xxx", trusted_connection="yes")

# 定义你要执行的查询语句
sql_query = "SELECT * FROM YOUR_TABLE_NAME"

# 获取转换好的DataFrame
result_df = data_fetcher.select(sql_query)

# 后续其他函数直接接收这个DataFrame就行
def process_data(df):
    # 示例：打印数据前5行
    print("数据预览：")
    print(df.head())

process_data(result_df)

这样修改后，你的类不仅能正确从SQL Server获取数据并转成DataFrame，还更灵活、更安全，后续其他函数直接接收返回的DataFrame就能继续处理啦~

内容的提问来源于stack exchange，提问作者Martinswe