Создайте столбец, выбрав случайные элементы из данного списка в pandas - PullRequest
1 голос
/ 07 апреля 2020

Создайте столбец, выбрав случайные элементы из заданного списка в pandas.

ID 
1
2
3
4
5

list = [0.1, 0.2, 0.3, 0.5, 0.8, 0.9]

Затем я хотел бы создать столбец с именем Вероятность, выбрав элемент случайным образом из приведенного выше списка с заменой.

Ожидаемый результат: (один из случайных выходов из 36 комбинаций)

ID    Probability
1     0.8
2     0.1
3     0.1
4     0.5
5.    0.8

1 Ответ

1 голос
/ 07 апреля 2020

Поскольку вы пометили numpy, вы можете использовать numpy .random.choice :

import numpy as np
import pandas as pd

l = [0.1, 0.2, 0.3, 0.5, 0.8, 0.9]

df = pd.DataFrame({'ID': range(1,6)})
df['Probability'] = np.random.choice(l, size=df.shape[0]) # By default replace=True

df
   ID  Probability
0   1          0.1
1   2          0.1
2   3          0.8
3   4          0.3
4   5          0.2
...