random.shuffle(分).在Python中拿(k)

Question

Python 中这行 Scala 的等价物是什么？

 random.shuffle(points).take(k)

我在Python

中好像没有找到shuffle的take方法

Answer 1

This 是你的一个选择，从 numpy 到随机洗牌 points

np.random.shuffle()

This 是您从 numpy 中随机选择 k 的一个选项 points

np.random.choice(points, size=k)

Answer 2

您可以使用 sample

从可迭代对象中挑选 k 个随机元素

import random

return random.sample(points, k)

此外，random 有 shuffle（但它正在变异），您可以使用切片代替 take:

copy = points[:]
random.shuffle(copy)
return copy[:k]