numpy.random.multinomial: есть ли способ сделать тест True или False? - PullRequest
0 голосов
/ 14 марта 2019

Используя 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, не похоже, что есть простой способ сделать это, и может быть проще использовать что-то еще. Буду признателен за некоторые мысли о том, как это сделать

1 Ответ

3 голосов
/ 14 марта 2019

Вы, вероятно, можете использовать np.random.choice, чтобы сделать что-то вроде:

df['T/F'] = np.random.choice([True, False], p=[0.4, 0.6], size=len(df))
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...