numpy.random.multinomial：是否可以进行True或False测试？

Question

使用numpy.random.multinomial，我想知道是否有可能遍历pandas数据框中的一列值并为它们分配状态True或False。例如，我在熊猫数据框中具有这些值。

col1 col2
240 0.0193549635804102
449 0.019230695266556665
355 0.01901116107244241
29  0.01886820933146161
301 0.017441394009229984

我有40％的可能性为True，而有60％的可能性为False，我想使用numpy.random.multinomial生成一个看起来像这样的数据框

col1 col2                 T/F
240 0.0193549635804102    True
449 0.019230695266556665  False
355 0.01901116107244241   True
29  0.01886820933146161   True
301 0.017441394009229984  False

查看numpy.random.multinomial的文档，似乎没有一种简单的方法可以执行此操作，并且使用其他方法可能更容易。我希望能有一些关于如何执行此操作的想法

Answer 1

您可能可以使用np.random.choice做类似的事情：

df['T/F'] = np.random.choice([True, False], p=[0.4, 0.6], size=len(df))