题目：
母牛从3~7岁初每年会生产1头小母牛，10岁后死亡（10岁任然存活），假设初始有一头刚出生的母牛，请问第n年有多少头母牛？（年从第一年开始计数）
注：第三年初会出生 第一头母牛，故第三年有两头母牛。
第五年初，第三年出生的母牛会生产，故第五年有五头母牛。
岁数是虚数
示例：
输入 输出
2 1
3 2
4 3
5 5
12 123
分析：
为计算第n年的牛数，我们需要知道每头牛的年龄。年龄不需要存储和更新，只记录每年有多少头小牛出生，即可推算得到。
创建列表，记录每年牛的总数和当年出生的小牛数[all_nums, birth_num]；
第n年新生小牛数 = 处在生育年龄段内小牛牛数总和
第n年的牛数 = 10年内出生的小牛总和
代码：
import numpy as np
# 输入要计算牛数的年份
n = int(input())
#n = 12
#创建列表，记录每年牛的总数和当年出生的小牛数[all_nums, birth_num]；
nums = []
# 初始化第1年小牛数：共1头牛，新出生1头牛
nums.append([1, 1])
nums = np.array(nums)
# 计算往后每年的牛数
for year in range(1, n):
#第n年新生小牛数 = 处在生育年龄段内小牛牛数总和
if year >= 7:
birth_num = sum(nums[-6:-1, 1]) #[:][-6:-1])
else:
birth_num = sum(nums[:-1, 1])
#第n年的牛数 = 10年内出生的小牛总和
if year >= 10:
all_nums = sum(nums[-9:, 1]) + birth_num
else:
all_nums = sum(nums[:, 1]) + birth_num
new = np.array([all_nums, birth_num])
nums = np.row_stack((nums, new))
print(nums[-1, 0])
#print(nums)
