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
相关产品推荐
相关产品推荐

