numpy:求解小游戏概率
甲和乙玩报数字的游戏,每人从1到20的整数中(含1和20)随机选择一个且同时给出数字,如果两个数字之差大于3,则甲赢;反之,则乙赢。问,甲赢的概率是多少?
#开发环境为jupyter notebook
import numpy as np
#生成数组
>>>n=20
>>>a=np.arange(1,n+1)
>>>print(a)
>>>[ 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20]
#扩充一个新数组,并利用numpy的广播功能,计算甲、乙两人所选数值的差额矩阵
>>>b=np.tile(a,[n,1]).T
>>>np.abs(a-b)
#计算差额矩阵中,所有差额大于3的个数,除以总个数(20*20=400),即为甲赢的概率
>>>np.sum(np.abs(a-b)>3)/n**2
>>>0.68