Django 得到一个随机对象

Question

我正在尝试从模型 A 中获取随机对象

目前，这段代码运行良好：

random_idx = random.randint(0, A.objects.count() - 1)
random_object = A.objects.all()[random_idx]

但我觉得这段代码更好：

random_object = A.objects.order_by('?')[0]

哪个是最好的？使用第一个代码删除对象可能出现问题？因为，例如，我可以有 10 个对象，但编号为 10 的对象不再存在？我是否误解了 A.objects.all()[random_idx] 中的某些内容？

Answer 1

刚刚在看这个。行：

random_object = A.objects.order_by('?')[0]

据报道已经关闭了许多服务器。

不幸的是，Erwans 代码在访问非顺序 ID 时导致错误。

还有另一种简单的方法：

import random

items = list(Product.objects.all())

# change 3 to how many random items you want
random_items = random.sample(items, 3)
# if you want only a single random item
random_item = random.choice(items)

这样做的好处是它可以无错误地处理非顺序 ID。

Answer 2

第二段代码是正确的，但速度可能会慢一些，因为在 SQL 中会生成一个 ORDER BY RANDOM() 子句，该子句会打乱整个结果集，然后在此基础上采用 LIMIT。

第一段代码仍然需要评估整个结果集。例如，如果您的 random_idx 接近最后一个可能的索引怎么办？

更好的方法是从您的数据库中选择一个随机 ID，然后选择它（这是一个主键查找，因此速度很快）。我们不能假设我们在1 和MAX(id) 之间的每个id 都可用，以防您删除了某些内容。所以以下是一个效果很好的近似值：

import random

# grab the max id in the database
max_id = A.objects.order_by('-id')[0].id

# grab a random possible id. we don't know if this id does exist in the database, though
random_id = random.randint(1, max_id + 1)

# return an object with that id, or the first object with an id greater than that one
# this is a fast lookup, because your primary key probably has a RANGE index.
random_object = A.objects.filter(id__gte=random_id)[0]

Answer 3

在以上所有方面都有所改进：

from random import choice

pks = A.objects.values_list('pk', flat=True)
random_pk = choice(pks)
random_obj = A.objects.get(pk=random_pk)

Answer 4

如何计算最大主键并获得随机pk？

《Django ORM Cookbook》一书比较了以下函数的执行时间，以从给定模型中获取随机对象。

from django.db.models import Max
from myapp.models import Category

def get_random():
    return Category.objects.order_by("?").first()

def get_random3():
    max_id = Category.objects.all().aggregate(max_id=Max("id"))['max_id']
    while True:
        pk = random.randint(1, max_id)
        category = Category.objects.filter(pk=pk).first()
        if category:
            return category

对一百万个数据库条目进行了测试：

In [14]: timeit.timeit(get_random3, number=100)
Out[14]: 0.20055226399563253

In [15]: timeit.timeit(get_random, number=100)
Out[15]: 56.92513192095794

见source。

看到这些结果后，我开始使用以下 sn-p：

from django.db.models import Max
import random

def get_random_obj_from_queryset(queryset):
    max_pk = queryset.aggregate(max_pk=Max("pk"))['max_pk']
    while True:
        obj = queryset.filter(pk=random.randint(1, max_pk)).first()
        if obj:
            return obj

到目前为止，只要有一个 id，它就可以完成这项工作。 请注意，如果您将模型 id 替换为 uuid 或其他内容，get_random3 (get_random_obj_from_queryset) 函数将不起作用。此外，如果删除了太多实例，while 循环会减慢进程。

Answer 5

另一种方式：

pks = A.objects.values_list('pk', flat=True)
random_idx = randint(0, len(pks)-1)
random_obj = A.objects.get(pk=pks[random_idx])

即使 pk 中存在较大间隙也可以工作，例如，如果您想在随机选择剩余对象之一之前过滤查询集。

编辑：固定调用 randint（感谢@Quique）。停止 arg 包含在内。

https://docs.python.org/3/library/random.html#random.randint

Answer 6

我正在与 Django 2.1.7、PostgreSQL 10 分享我的最新测试结果。

students = Student.objects.all()
for i in range(500):
    student = random.choice(students)
    print(student)

# 0.021996498107910156 seconds

for i in range(500):
    student = Student.objects.order_by('?')[0]
    print(student)

# 0.41299867630004883 seconds

使用 random.choice() 进行随机获取似乎快了大约 2 倍。

Answer 7

您可以使用“随机”模块中的“选择”

from .models import MyModel
from random import choice    

MyRandomChoice = choice(MyModel.objects.all())