Python在列表中執行計算並將其添加到列表中

Question

我試圖在Python中運行一個模型，該模型將對列表中的3個項目應用某些計算，然後繼續將這些計算應用於一組迭代次數。

下面的代碼是我到目前爲止。我是否需要使用諸如modulo或index（）來保持if/elif繼續增長？或者我會更好地融入功能？

# Nonsense Model for Fish numbers ### 
''' Fish numbers, in 100,000 tonnes in coastal waters. Those in the North Sea 
    increase by 10% every year. 
Those in the Irish Sea increase by 15%. Those in the Channel are currently decreasing 
the difference by 5%. Also want a total for each year for 10 year prediction.''' 
#master list 
fish = [] 
#ask questions 
northFish = int(input('How many fish in first year the North Sea?')) 
irishFish = int(input('How many fish in first year the Irish Sea?')) 
channelFish = int(input('How many fish in first year in The Channel?')) 

# add to the list 
fish.extend((northFish,irishFish,channelFish)) 
# display the start 
print (fish) 
year = int(input('how many years would you like to model?')) 
#run the model 

for i in range(1,year): 
    newFish = [] 
    if i == 0: 
     n = fish[0] * 1.1 
     newFish.append(n) 
    elif i == 1: 
     ir = fish[1] * 1.15 
     newFish.append(ir) 
    elif i == 2: 
     c = fish[2] * 0.95 
     newFish.append(c)  

    fish.extend(newFish) # add generated list to the master list  
    #total = n + i + c  
    #print('The total for year', i, 'is', total) 
print(fish)  

Answer 1

編輯：這是你如何也可以保存每年的趨勢。列表fish包含當年的人口，列表fish_trend包含每年的魚羣。例如fish_trend[5][1]將在5年後返回愛爾蘭魚羣。

# Nonsense Model for Fish numbers ### 
''' Fish numbers, in 100,000 tonnes in coastal waters. Those in the North Sea 
    increase by 10% every year. 
Those in the Irish Sea increase by 15%. Those in the Channel are currently decreasing 
the difference by 5%. Also want a total for each year for 10 year prediction.''' 
# master list 
fish = [] 
# ask questions 
northFish = int(input('How many fish in first year the North Sea?')) 
irishFish = int(input('How many fish in first year the Irish Sea?')) 
channelFish = int(input('How many fish in first year in The Channel?')) 

# add to the list 
fish = [northFish, irishFish, channelFish] 
# display the start 
print(fish) 
year = int(input('how many years would you like to model?')) 
# run the model 

factors = (1.1, 1.15, 0.95) 

# initialize fish_trend with population of first year 
fish_trend = [tuple(fish)] 

# iterate through years 
for _ in range(year): 
    # iterate through fish types 
    for i in range(len(fish)): 
     fish[i] *= factors[i] 

    # append current population to fish_trend 
    fish_trend.append(tuple(fish)) 

# you can use pprint for better visualization 
import pprint 
pprint.pprint(fish_trend) 

# if you want the sum of the population per year, you can do that easily by using this: 
sum_per_year = [sum(f) for f in fish_trend] 

print(sum_per_year)