Question

好的，所以对于课堂我们有这个问题，我们需要能够输入一个单词，并且从给定的文本文件（wordlist.txt）中，将使用文件中找到的该单词的任何字谜制作一个列表。

到目前为止我的代码看起来像这样：

def find_anagrams1(string):
"""Takes a string and returns a list of anagrams for that string from the wordlist.txt file.

string -> list"""
anagrams = []

file = open("wordlist.txt")
next = file.readline()
while next != "":
    isit = is_anagram(string, next)
    if isit is True:
        anagrams.append(next)
    next = file.readline()
file.close()

return anagrams

每次我尝试运行该程序时，它只会返回一个空列表，尽管事实上我知道存在一个字谜。关于什么是错的任何想法？

P.S。 is_anagram函数如下所示：

def is_anagram(string1, string2):
"""Takes two strings and returns True if the strings are anagrams of each other.

list,list -> string"""
a = sorted(string1)
b = sorted(string2)
if a == b:
    return True
else:
    return False

我正在使用Python 3.4

Answer 1

问题是您使用的是readline功能。来自文档：

file.readline = readline(...)
readline([size]) -> next line from the file, as a string.

Retain newline.  A non-negative size argument limits the maximum
number of bytes to return (an incomplete line may be returned then).
Return an empty string at EOF.

这里的关键信息是＆＃34;保留换行符＆＃34;。这意味着如果你有一个包含单词列表的文件，每行一个，每个单词将返回一个终端换行符。所以当你打电话时：

next = file.readline()

您未获得example，而您正在获取example\n，因此这将永远不会与您的输入字符串匹配。

一个简单的解决方案是在从文件读取的行上调用strip()方法：

next = file.readline().strip()
while next != "":
    isit = is_anagram(string, next)
    if isit is True:
        anagrams.append(next)
    next = file.readline().strip()
file.close()

但是，此代码存在一些问题。首先，file是一个变量的可怕名称，因为这将掩盖python file模块。

不是反复调用readline()，而是利用open文件是一个产生文件行的迭代器这一事实，你最好不要这样做：

words = open('wordlist.txt')
for word in words:
    word = word.strip()
    isit = is_anagram(string, word)
    if isit:
      anagrams.append(word)
words.close()

此处还请注意，由于is_anagram返回True或False，因此不需要将结果与True或False（例如if isit is True）进行比较。您可以单独使用返回值。

Answer 2

哎呀，不要用于循环：

import collections

def find_anagrams(x):
    anagrams = [''.join(sorted(list(i))) for i in x]
    anagrams_counts = [item for item, count in collections.Counter(anagrams).items() if count > 1]
    return [i for i in x if ''.join(sorted(list(i))) in anagrams_counts]

Answer 3

这是另一种解决方案，我认为它非常优雅。这在 O(n * m) 中运行，其中 n 是单词数，m 是字母数（或平均字母数/单词数）。

# anagarams.py

from collections import Counter
import urllib.request


def word_hash(word):
  return frozenset(Counter(word).items())

def download_word_file():
  url = 'https://raw.githubusercontent.com/first20hours/google-10000-english/master/google-10000-english-no-swears.txt'
  urllib.request.urlretrieve(url, 'words.txt')

def read_word_file():
  with open('words.txt') as f:
    words = f.read().splitlines()
  return words


if __name__ == "__main__":
  # downloads a file to your working directory
  download_word_file()

  # reads file into memory
  words = read_word_file()

  d = {}

  for word in words:
    k = word_hash(word)
    if k in d:
      d[k].append(word)
    else:
      d[k] = [word]

  # Prints the filtered results to only words with anagrams
  print([x for x in d.values() if len(x) > 1])

查找文件中给定单词的字谜

3 个答案: