python list comprehension无法正常工作

时间:2018-11-11 21:26:37

标签: python list-comprehension

我想编写一种方法,该方法将根据传递的参数从文件中返回一个单词。但是,如果没有适合该参数的单词,我什么也不想返回。因此,在我的文件中,最高的单词有97分。但是,如果我通过分数98,则会显示有关索引的错误。我有这样的东西:

main.py

from option import Option
import argparse

parser = argparse.ArgumentParser()
group = parser.add_mutually_exclusive_group()

group.add_argument( "--score", "-s", help="Find word for given score", type=int)

option = Option()
if args.score:
    option.word_from_score(args.score)

option.py

import random
class Option():
    def __init__(self):
        self.file = [line.rstrip('\n').upper() for line in open('dictionary.txt', "r")] 

    SCRABBLES_SCORES = [(1, "E A O I N R T L S U"), (2, "D G"), (3, "B C M P"),
                (4, "F H V W Y"), (5, "K"), (8, "J X"), (10, "Q Z"), (11, "Ą Ć Ę Ł Ń Ó Ś Ź Ż")]
    global LETTER_SCORES 
    LETTER_SCORES = {letter: score for score, letters in SCRABBLES_SCORES
             for letter in letters.split()}

    def word_from_score(self,score):
        print(random.choices([word for word in self.file if sum([LETTER_SCORES[letter] for letter in word ]) == score]))

此方法返回单词,但不处理错误。所以我尝试了这个:

def word_from_score(self,score):
    print(random.choices([(word if sum([LETTER_SCORES[letter] for letter in word ]) == score else "") for word in self.file]))

但是在这种情况下,我传递的每个参数都会返回“”。这种方法在哪里出错?

[EDIT]例如,我从命令行运行程序,并且:

python main.py -f

返回97,因为这是文件中某些单词的分数。因此,如果我运行其他方法:

pythom main.py -s 97

从文件中返回单词,该单词具有此分数。而且有效。但是,如果我以98来作为参数,它将无法正常工作,因为在文件中此分数没有任何意义。现在我想处理这种情况,返回“”

1 个答案:

答案 0 :(得分:0)

当前,您的列表压缩功能正在构建一个与得分匹配的单词列表,但其中还包括与得分不匹配的所有单词的空字符串。您想要这样的东西:

def word_from_score(self,score):
    valid_words = [word for word in self.file if sum([LETTER_SCORES[letter] for letter in word ]) == score]
    if len(valid_words) != 0:
        print(random.choices(valid_words))
    else:
        print('')