You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Python函数如何返回多行数据?当前仅返回单条记录

解决Python函数无法返回多行数据的问题

你的代码里,return row在循环第一次执行时就会立即终止函数并返回当前行,所以只能拿到第一条记录。以下是不用pandas的可行方案:

  • 直接返回全部结果:fetchall()本身就会返回查询结果的所有行组成的列表,没必要循环,直接返回即可:
def extract(self):
    return self.oracle_conn().execute('select * from employees').fetchall()
  • 使用生成器逐行返回:如果数据量很大,不想一次性加载所有数据到内存,可以用yield代替return,生成器会逐行输出数据:
def extract(self):
    cursor = self.oracle_conn().execute('select * from employees')
    for row in cursor:
        yield row

调用时可以通过循环遍历获取每一行:

for employee in your_instance.extract():
    print(employee)
  • 收集结果到列表后返回:如果需要对每行数据做额外处理(比如格式转换),可以先把所有行收集到列表里,处理完再返回:
def extract(self):
    cursor = self.oracle_conn().execute('select * from employees')
    result_list = []
    for row in cursor:
        # 这里可以添加对row的自定义处理逻辑
        processed_row = row  # 示例:直接保留原行,可替换为处理逻辑
        result_list.append(processed_row)
    return result_list

内容的提问来源于stack exchange,提问作者Jim Macaulay

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.19 20:35:16