Django中指定k值后choices函数仍返回随机数量对象问题

问题根源 #

你用random.choices()选取6个pk，但这个函数是有放回的随机抽样——同一个pk可能被选中多次。而Django的filter(pk__in=...)会自动剔除重复对象，所以当抽样结果里有重复pk时，最终返回的albums数量就会少于6，导致数量不固定。

解决方案 #

方案1：无放回抽样（推荐，保证不重复且数量固定） #

改用random.sample()做无放回抽样，确保选出的pk都是唯一的。同时处理专辑数量不足6的情况，避免抛出异常：

import random

def homeview(request):
    pks = list(Album.objects.values_list('pk', flat=True))
    # 如果专辑总数不足6，就取全部
    select_count = min(6, len(pks))
    random_pks = random.sample(pks, k=select_count)
    albums = Album.objects.filter(pk__in=random_pks)

    return render(
        request,
        'home/home.html',
        {'albums': albums},
    )

方案2：允许返回重复对象（特殊业务场景） #

如果业务允许出现重复专辑，可以根据抽样的pk列表逐个获取对象，保留重复项：

import random

def homeview(request):
    pks = list(Album.objects.values_list('pk', flat=True))
    random_pks = random.choices(pks, k=6)
    albums = []
    for pk in random_pks:
        # 用filter+first避免对象被删除时抛出异常
        album = Album.objects.filter(pk=pk).first()
        if album:
            albums.append(album)

    return render(
        request,
        'home/home.html',
        {'albums': albums},
    )

方案3：Django原生随机查询（小数据量适用） #

直接用Django的order_by('?')实现随机排序后取前6个，代码更简洁，但数据量大时性能较差（会触发全表排序）：

def homeview(request):
    albums = Album.objects.order_by('?')[:6]

    return render(
        request,
        'home/home.html',
        {'albums': albums},
    )

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