python怎么把程序封装成函数_怎么把下面的Python代码封装成函数？

该楼层疑似违规已被系统折叠 隐藏此楼查看此楼

下面的两段代码是经过简化后的，现在想封装成一个函数，传入一个list p，然后返回list

A，因为p的长度是不定的，一时想不起来如何抽象，似乎应该用两个循环？

import random

p=[0.85, 0.92, 0.99, 1]

A=[]

for i in range(100):

....r= random.random()

....if r < p[0]:

........x=0

....elif r < p[1]:

........x=1

....elif r < p[2]:

........x=2

....else:

........x=3

....A.append(x)

print A

#----------------------------------

p=[0.25, 0.5, 0.75, 0.95, 1]

A=[]

for i in range(100):

....r= random.random()

....if r < p[0]:

........x=0

....elif r < p[1]:

........x=1

....elif r < p[2]:

........x=2

....elif r < p[3]:

........x=3

....else:

........x=4

....A.append(x)

print A