我有四个字符串的列表。然后在Pandas数据框中,我想创建一个变量,从该列表中随机选择一个值,并将其分配给每一行。我使用的是numpy的随机选择,但是阅读他们的文档,没有任何种子选项。如何为随机分配指定随机种子,以便每次随机分配都相同?
service_code_options = ['899.59O', '12.42R', '13.59P', '204.68L']
df['SERVICE_CODE'] = [np.random.choice(service_code_options ) for i in df.index]
最佳答案
您需要在 numpy.random.seed
之前定义它,也不需要列表理解,因为可以将 numpy.random.choice
与size
参数一起使用:
np.random.seed(123)
df = pd.DataFrame({'a':range(10)})
service_code_options = ['899.59O', '12.42R', '13.59P', '204.68L']
df['SERVICE_CODE'] = np.random.choice(service_code_options, size=len(df))
print (df)
a SERVICE_CODE
0 0 13.59P
1 1 12.42R
2 2 13.59P
3 3 13.59P
4 4 899.59O
5 5 13.59P
6 6 13.59P
7 7 12.42R
8 8 204.68L
9 9 13.59P