Python，我在重複自己很多，當涉及到循環，並且必須有更好的方式

Question

可以說我有三個列表，我需要遍歷它們並做一些內容。

這三個列表是streaks_0,streaks_1和streaks_2。對於每個列表，我需要使用特定於每個列表的不同值。例如，streak_0_num0s在streaks_1 for循環中將不起作用。

有沒有辦法讓這三個循環成爲一個或至少一個方法來清理呢？

for number in streaks_0: 
    if number == 0: 
     streak_0_num0s += 1 
    elif number != 0: 
     streak_0_sum += number 
streak_0_average = (streak_0_sum/(len(streaks_0) - streak_0_num0s)) 

for number in streaks_1: 
    if number == 0: 
     streak_1_num0s += 1 
    elif number != 0: 
     streak_1_sum += number 
streak_1_average = (streak_1_sum/(len(streaks_1) - streak_1_num0s)) 

for number in streaks_2: 
    if number == 0: 
     streak_2_num0s += 1 
    elif number != 0: 
     streak_2_sum += number 
streak_2_average = (streak_2_sum/(len(streaks_2) - streak_2_num0s)) 

Answer 1

爲什麼不使用函數？

def get_average(streaks): 
    streak_0_num0s = 0 
    streak_0_sum = 0 

    for number in streaks: 
     if number == 0: 
      streak_0_num0s += 1 
     elif number != 0: 
      streak_0_sum += number 
    streak_0_average = (streak_0_sum/(len(streaks) - streak_0_num0s)) 
    print(streak_0_average) 

get_average(streaks01) 
get_average(streaks02) 
get_average(streaks03) 

Answer 2

您的代碼可以與下面的一個類似的功能很容易簡化爲：

def calculate_avg(lst): 
    return sum(lst)/(len(lst)-lst.count(0)) 

或這一個，如果你喜歡：

def calculate_avg(lst): 
    return sum(lst)/len([l for l in lst if l != 0]) 

，這裏是一個小的使用例子：

streaks = [ 
    [1, 2, 3, 0, 0, 0, 0], 
    [0, 0, 0, 4, 5, 6, 0], 
    [0, 0, 6, 7, 8, 0, 0] 
] 

for index, streak in enumerate(streaks): 
    print("avg(streak{})={}".format(str(index).zfill(2), calculate_avg(streak))) 

Answer 3

寫一個函數離子，就可以多次調用：

def calculate_average(values): 
    non_zeros = 0 
    sum = 0 

    for value in values: 
     if value != 0: 
      sum += value 
      non_zeros += 1 
    return sum/non_zeros 

streak_0_average = calculate_average(streaks_0) 
streak_1_average = calculate_average(streaks_1) 
streak_2_average = calculate_average(streaks_2) 

Answer 4

正如別人已經說過：當你發現你自己重複了很多遍：嘗試創建可重複使用的功能。

但是，環顧一下並查看是否有人已經實現了這樣的功能總是一個好主意。在你的情況下，你可以使用numpy.mean（numpy是第三方模塊）或statistics.mean（statistics是python 3.4+中的內置模塊）。

他們不默認做的是不包括零，所以你必須自己做的唯一的事情：

import numpy as np 

def average(streaks): 
    streaks = np.asarray(streaks) 
    streaks_without_zeros = streaks[streaks != 0] 
    return np.mean(streaks_without_zeros) 

streaks_0 = [1, 2, 3, 4, 0, 1, 2, 3] 
print(average(streaks_0)) # 2.2857142857142856 

或：

import statistics 

def average(streaks): 
    streaks_without_zeros = [streak for streak in streaks if streak != 0] 
    return statistics.mean(streaks_without_zeros) 

streaks_0 = [1, 2, 3, 4, 0, 1, 2, 3] 
print(average(streaks_0)) # 2.2857142857142856 

當然