如何在使用Python的numpy随机选择时指定随机种子？

Question

我有四个字符串的列表。然后在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]

Answer 1

您需要在by之前定义它numpy.random.seed，也不需要列表理解，因为可以numpy.random.choice与parameter一起使用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