在列表中添加项目 - Python

Question

我试图编写一个练习的代码，虽然它的工作方式应该如此，但它看起来很可怕，我想知道是否有人可以帮助我删除任何不必要的东西，并且可能只是结合一些功能？

下面是使用功能的一个例子：

choices([['YES', 'NO', 'YES', 'YES'], ['NO', 'NO', 'YES', 'NO'], ['YES', 'YES', 'YES', 'YES']]) 

列表中的每个列表有四个是/否选择（下面列出的四个选项指数为好，如绿色，红色，蓝色，黄色;但它不一定是四）。列表中的列表数量是多少人投票。

i = 0 
num = 0 
total_num = 0 
num_choices = len(INDICES) 
choices_index = 0 
choices_votes = [] 
choices_votes_num = [] 
index = 0 
total_choices = [] 
winning_choice = '' 
winning_index = 0 
while i < len(parameter): 
    while num < num_choices: 
     for item in parameter: 
      choices_votes.append(item[num]) 
     num += 1 
    i += 1 
while total_num < len(choices_votes): 
    if choices_votes[total_num] == 'YES': 
     choices_votes_num.append(1) 
     total_num += 1 
    elif choices_votes[total_num] == 'NO': 
     choices_votes_num.append(0) 
     total_num += 1 
while choices_index < len(choices_votes_num): 
    count = int(len(choices_votes_num)/num_choices) 
    total = 0 
    total = sum(choices_votes_num[choices_index:(choices_index + count)]) 
    total_choices.append(total) 
    choices_index = choices_index + count 
for score in total_choices: 
    winning_index = max(total_choices) 
    winning_choice = INDEX_TO_NAME[total_choices.index(winning_index)] 
return winning_choice, total_choices 

INDEX_TO_NAMEINDEX_TO_NAME只是一个字典，它将索引连接到选项（颜色）。

基本上，代码应该将每个yes计为1分，每个no计为零分，将每个可用选项的总积分加起来，然后返回总计和获胜者。

Answer 1

让我们开始：

c = [['YES', 'NO', 'YES', 'YES'], 
    ['NO', 'NO', 'YES', 'NO'], 
    ['YES', 'YES', 'YES', 'YES']] 

INDICES = ['red', 'green', 'blue', 'yellow'] 

既然你使用INDICES反正你的来源调查，我们可以假设，答案总是排队与INDICES数量。

我们可以利用zip重组这样的数据：

zip(*c) 
#[('YES', 'NO', 'YES'), 
# ('NO', 'NO', 'YES'), 
# ('YES', 'YES', 'YES'), 
# ('YES', 'NO', 'YES')] 

这扩大了答案，争论和重组他们的实际指标进行分组。因此，第一个指标是“红”，二是“绿色”等

现在我们可以再次压缩，将其与指数相结合：

results = zip(INDICES, zip(*c)) 
#[('red', ('YES', 'NO', 'YES')), 
# ('green', ('NO', 'NO', 'YES')), 
# ('blue', ('YES', 'YES', 'YES')), 
# ('yellow', ('YES', 'NO', 'YES'))] 

，我们可以遍历results，简单地计算的出现'是'：

totals = [(ind, answers.count('YES')) for ind,answers in results] 
#[('red', 2), ('green', 1), ('blue', 3), ('yellow', 2)] 

所以这里我们有总数。我们可以把它传递到调用max把它告诉我们哪一个是赢家：

max(totals, key=lambda x: x[1]) 
#('blue', 3) 

max默认情况下将首先寻找的指数，所以我们可以把它key功能指示它拉而是索引1。它表明我们蓝色是赢家。

更有效率，我们实际上可以发电机传递到max：

totals = ((ind, answers.count('YES')) for ind,answers in results) 
#<generator object <genexpr> at 0x102581fa0> 
max(totals, key=lambda x: x[1]) 
#('blue', 3) 

最后一个语句可以这样写：

max(((ind, answers.count('YES')) for (ind,answers) in zip(INDICES, zip(*c))), 
    key=lambda x: x[1]) 

我知道这个答案引入了一些列表理解和lambda（以及generator），但是您需要一种清理方法，这是您可以使用的工具的一个示例。希望这会给你你需要的帮助！

注意：此答案不考虑平局。但我确信你可以从这里工作，将这些例子整合到你的实际代码中