更改`voting_approval`的输出

Question

我正在尝试编写一个接受参数的函数，该参数是一个由4个元素组成的列表，这些列表表示单次骑行的批准选票；内部列表元素的顺序对应于各方在名为PARTY_INDICES的各方列表中的顺序。

投赞成票最多的政党获胜。

它应该返回一个2元组，其中第一个元素是获胜方的名字，第二个元素是一个四元素列表，其中包含每个政党的赞成票数量。列表元素的顺序与PARTY_INDICES中各方的顺序相对应。

我的代码是：

def voting_approval(approval):
    parties = ['NDP','GREEN','LIBERAL','CPC']
    values = [0,0,0,0]
    for decision in approval:
        for no, item in enumerate(decision):
            if item == 'Yes':
                values[no] += 1
                total = [(values) for x in zip(approval)]

    return (parties[values.index(max(values))], total)

如果我尝试：

voting_approval([['Yes', 'No', 'Yes', 'No'],['No', 'No', 'Yes', 'No']])

它会吐出：

('LIBERAL', [[1, 0, 2, 0], [1, 0, 2, 0]])

但我希望结果是：

('LIBERAL', [1, 0, 2, 0])

Answer 1

没有进行过广泛的测试，但您可以尝试：

In [7]: def voting_approval(approval):
   ...:     parties = ['NDP', 'GREEN', 'LIBERAL', 'CPC']
   ...:     values = [sum(1 if i == 'Yes' else 0 for i in group) for group in zip(*approval)]
   ...:
   ...:     return (parties[values.index(max(values))], values)
   ...:

In [8]: voting_approval([['Yes', 'No', 'Yes', 'No'],['No', 'No', 'Yes', 'No']])
Out[8]: ('LIBERAL', [1, 0, 2, 0])

这将在开始时将你的列表压缩在一起，对Yes求和并创建一个带有总数的新列表，然后用它来提取所需的信息。如果这不是你想要的，我很乐意调整。

Answer 2

您已经在values中存储了每一方的“YES”数量。只需按如下所示更改代码即可。

def voting_approval(approval):
    parties = ['NDP','GREEN','LIBERAL','CPC']
    values = [0,0,0,0]
    for decision in approval:
        for no, item in enumerate(decision):
            if item == 'Yes':
                values[no] += 1

    return (parties[values.index(max(values))], values)

Answer 3

您不需要这个total变量。values已经是你想要的了。

def voting_approval(approval):
    parties = ['NDP','GREEN','LIBERAL','CPC']
    values = [0,0,0,0]
    for decision in approval:
        for no, item in enumerate(decision):
            if item == 'Yes':
                values[no] += 1

    return (parties[values.index(max(values))], values)

approvals = [
    ['Yes', 'No', 'Yes', 'No'],
    ['No', 'No', 'Yes', 'No'],
]

print voting_approval(approvals)

matt@wraith:~/Dropbox/Public/StackOverflow$ python 13638424.py 
('LIBERAL', [1, 0, 2, 0])