python 递归函数 古典问题兔子

‘’’
古典问题：有一对兔子，从出生后第3个月起每个月都生一对兔子，小兔子长到第三个月后每个月又生一对兔子，假如兔子都不死，
问每个月的兔子总数为多少？
‘’’
def rab(month):
‘’’
该函数是递归函数的写法，返回兔子的个数
:param month: 这是第几个月
:return: 兔子的个数
‘’’
if month == 1:
return 1
elif month == 2:
return 1
else:
count = rab(month-1)+rab(month-2)
return count

print(rab(3))
#输出2