我对python有点陌生,所以我想知道是否有更有效的方法来完成某件事。基本上,我需要根据随机数(0或1)创建一个来自其他两个数组之一的值的数组,目前,使用for循环非常容易实现,但是我只是好奇是否存在更优雅的/类似于python的方式来完成此操作,对于python的设计而言似乎太笨拙了:
import random
xySet = ['x', 'y']
xP = [10.1, 11.2, 12.3]
yP = [12.5, 13.2, 14.1]
nObser = 10
x = []
p = []
randVals = [random.randint(0,1) for i in range(nObser)]
print randVals
for i in range(nObser):
x.append(xySet[randVals[i]])
if randVals[i]:
p.append(xP[random.randint(0,2)])
else:
p.append(yP[random.randint(0,2)])
print x
print p
这给了我期望的正确输出:
[1, 1, 1, 0, 1, 1, 1, 0, 1, 0]
['y', 'y', 'y', 'x', 'y', 'y', 'y', 'x', 'y', 'x']
[12.3, 11.2, 10.1, 13.2, 10.1, 11.2, 12.3, 14.1, 10.1, 13.2]
谢谢!
您可以使用一对列表推导
>>> from random import choice
>>> x = [choice(xySet) for _ in range(nObser)]
>>> p = [choice(xP) if v == 'x' else choice(yP) for v in x]
>>> x
['y', 'y', 'y', 'x', 'x', 'x', 'y', 'x', 'x', 'x']
>>> p
[14.1, 13.2, 14.1, 10.1, 10.1, 12.3, 13.2, 12.3, 11.2, 10.1]
在第一个列表理解中,您可以random.choice
用来随机选择'x'
或'y'
。
然后,在第二个列表理解中,您遍历x
列表并random.choice
从适当的列表中采样(再次使用)。
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句