我想编写一种方法,该方法将根据传递的参数从文件中返回一个单词。但是,如果没有适合该参数的单词,我什么也不想返回。因此,在我的文件中,最高的单词有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来作为参数,它将无法正常工作,因为在文件中此分数没有任何意义。现在我想处理这种情况,返回“”
答案 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('')