Django如何将外键的指定字段加载到QuerySet结果中?
问题原因
你之前的写法存在两处错误:
- 关联字段使用错误:
painter_set是Painter反向查询关联Painting的默认字段名,从Painting正向查询关联的Painter,直接使用外键字段名painter即可 prefetch_related的作用是提前加载关联对象减少查询次数,无法直接把关联字段的值写到Painting的查询结果字段中
实现方案
提示:因为是正向外键关联,使用
select_related("painter")可以让Django执行联表查询,只会触发1次SQL请求,避免导出大量数据时出现N+1查询性能问题。
方案1:注解字段+排除多余字段(Django 2.1+ 推荐,无需手动列全Painting字段)
用annotate将关联的画家名字段直接注入到查询集中,导出时排除原有的painter外键字段即可:
from django.db.models import F import pandas as pd # 注解添加painter_name字段 paintings = Painting.objects.select_related("painter").annotate( painter_name=F("painter__name") ).all() # 导出时排除painter字段(即原有的painter_id) df = pd.DataFrame(list(paintings.values().exclude("painter"))) df.to_csv('blah.csv')
方案2:指定查询字段+重命名(兼容所有Django版本)
如果你的Django版本低于2.1,不支持注解别名,可以先提取Painting的所有字段,排除外键字段后手动加入关联的画家名字段:
import pandas as pd # 提取Painting所有非外键字段,不用手动枚举30个字段 painting_fields = [f.name for f in Painting._meta.fields if f.name != "painter"] # 跨表查询所有需要的字段 paintings = Painting.objects.select_related("painter").values( *painting_fields, "painter__name" ) # 转DataFrame后修改列名符合要求 df = pd.DataFrame(list(paintings)) df.rename(columns={"painter__name": "painter_name"}, inplace=True) df.to_csv('blah.csv')
两种方案最终导出的CSV都会包含Painting的全部30个业务字段,以及painter_name字段,不会返回多余的painter_id。
内容的提问来源于stack exchange,提问作者user1367204
相关产品推荐
相关产品推荐

