通过从熊猫的给定列表中选择随机元素来创建列

时间:2020-04-07 13:35:41

标签: pandas numpy pandas-groupby

通过从熊猫的给定列表中选择随机元素来创建列。

ID 
1
2
3
4
5

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

然后,我想通过从上面的列表中随机选择一个元素进行替换来创建一个名为Probability的列。

预期输出:(36个组合中的随机输出之一)

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

1 个答案:

答案 0 :(得分:1)

自从您标记了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