如何避免编码 if 语句两次

Question

我主要是自己编程，所以没有人检查我的代码。我觉得我已经养成了一堆坏习惯。

我在此处粘贴的代码有效，但我想听听其他一些解决方案。

我创建了一个名为teams_shots 的字典。我遍历了一个 pandas 数据框，该数据框在一行中包含客队和主队的名称。我想跟踪出现在数据框中的每个团队的投篮情况。这就是为什么我检查home_team_name 或away_team_name 在字典中是否没有条目，如果有，我创建一个。

for index,match in df.iterrows():
    if match['home_team_name'] not in teams_shots:
        #we have to setup an entry in the dictionary
        teams_shots[match['home_team_name']]=[]
        teams_shots[match['home_team_name']].append(match['home_team_shots'])
        home_shots_avg.append(None)
    else:
        home_shots_avg.append(np.mean(teams_shots[match['home_team_name']]))
        teams_shots[match['home_team_name']].append(match['home_team_shots'])

    if match['away_team_name'] not in teams_shots:
        teams_shots[match['away_team_name']]=[]
        teams_shots[match['away_team_name']].append(match['away_team_shots'])
        away_shots_avg.append(None)
    else:
        away_shots_avg.append(np.mean(teams_shots[match['away_team_name']])) 
        teams_shots[match['away_team_name']].append(match['away_team_shots'])

正如您所见，几乎相同的代码被编写了两次，这并不是良好编程的标志。我曾考虑在 if 语句中使用 or 运算符，但可能已经创建了一个条目，我会截断它。任何想法如何更好地编写此代码。

Answer 1

我会使用get 作为快速查找。它不会抛出KeyErrors 并且默认的None 在真实性中充当False

for index, match in df.iterrows():
    home, away, home_shots, away_shots = match['home_team_name'],
                           match['away_team_name'],
                           match['home_team_shots'],
                           match['away_team_shots']


    if not teams_shots.get(home):
        # No need to separately allocate the array
        teams_shots[home] = [home]
        home_shots_avg.append(None)
    else:
        home_shots_avg.append(np.mean(teams_shots[home_shots]))

    if not teams_shots.get(away):
        teams_shots[away] = [away]
        away_shots_avg.append(None)
    else:
        away_shots_avg.append(np.mean(teams_shots[away_shots]))

Answer 2

在这种情况下，我认为额外的for 循环应该可以解决问题：

for index,match in df.iterrows():
        for name, shots in {'home_team_name':'home_team_shots',
                            'away_team_name':'away_team_shots'}:

            if match[name] not in teams_shots:
                #we have to setup an entry in the dictionary
                teams_shots[name]=[]
                teams_shots[name].append(match[shots])
                home_shots_avg.append(None)
             else:
                home_shots_avg.append(np.mean(teams_shots[name]))

但可能有办法以矢量化方式处理此问题。