Python3：使用 CSV 自动完成文本

Question

我想做的是一个自动完成脚本，它使用一个 csv 文件，您可以从用户输入中获得最有可能的建议，使用 csv 的第二行（它说明给定单词的出现次数）确定可能性。

到目前为止，我已经在导入 CSV 后测试了两种方法，第一种是将其放入排序列表（最高数字优先）。然后我将循环并输出第一个遇到的以用户输入开头的单词。然而，这个想法从根本上存在缺陷，因为我找不到根据数字的实际值对列表进行排序的解决方案，只有开始的数字。例如。数字九 (9) 的排名高于 7184。

thelist = open('alphabetical.csv', 'r')
csv1 = csv.reader(thelist, delimiter = ',')
sortedlist = sorted(csv1, key = lambda x: x[1], reverse=True)

一个 sn-p 可能看起来像： [['with', '1671'], ['seldom', '2'], ['green', '246']] - 等等。

这是函数当前的样子：

def main():
"""Initialize main loop."""
word = ""

while word != "q":
    word = input("Type word: ").lower()
    print("Autocompletion finished: ", autocomplete())

def autocomplete():
"""Return autocomplete suggestions."""
filtered_words = filter(lambda x: x.startswith(word), sortedlist)
return filtered_words

输入后的结果是：

<filter object at 0x7fbfb27ca860>

我知道代码有缺陷，而且还有更多问题，所以输出不是它所需要的对我来说并不奇怪。

从当前函数获得给定输出后，我担心无法使用列表，因此我目前正在尝试使用字典，但我最不想做的就是在实际上一开始就在正确的轨道上。因此，我向您寻求有关如何解决此问题的任何可能的指导和建议。

谢谢。

Answer 1

首先，为什么要排序然后创建一个新列表？

sort = sorted(csv1, key = lambda x: x[1])
for row in sort:
    sortedlist.append(row)

可能

sortedlist = sorted(csv1, key = lambda x: x[1])

您的其他错误：filter 在 python 3 中不再返回列表，它是懒惰地评估。您必须强制对其进行列表迭代。

如果您这样做，您会意识到还有另一个错误，因为startswith 适用于 行，而不适用于字符串。

所以快速修复将是：

filtered_words = list(filter(lambda x: x[0].startswith(word), sortedlist))

这将返回行，而不是单词，顺便说一句。

但是使用list + filter + lambda 变得繁琐且性能不佳。更喜欢简单的列表理解：

filtered_words = [x for x in sortedlist if x[0].startswith(word)]

也返回行。要仅获取可以对两个字段使用解包并丢弃数字的单词

filtered_words = [x for x,_ in sortedlist if x.startswith(word)]

或者您可以从一开始就丢弃这些数字：

sortedlist = [x[0] for x in sorted(csv1, key = lambda x: x[1])]

这会让你改变过滤器：

filtered_words = [x for x in sortedlist if x.startswith(word)]

这种方法很简单，但正如 cmets 中所述，它不是最有效的，因为它意味着线性搜索（并且没有利用列表已排序的事实）