numpy.random.choice坑

numpy.random.choice 默认是有放回。其中有个 replace 的参数控制，默认是 True。如划分数据集用到，注意重复。

Code

import numpy as np

a = np.arange(5)
print(a)
for _ in range(3):
    b = np.random.choice(a, 3)  # 有放回
    c = np.random.choice(a, 3, replace=False)  # 无放回
    print("\nb:", b)
    print("c:", c)