Python：将文件中的单词加载到集合中

Question

我有一个简单的文本文件，其中包含数千个单词，每个单词都在自己的行中，例如，

aardvark
hello
piper

我使用以下代码将单词加载到集合中（我需要单词列表来测试成员资格，因此集合是我选择的数据结构）：

my_set = set(open('filename.txt'))

上面的代码生成一个包含以下条目的集合（每个单词后跟一个空格和换行符：

("aardvark \n", "hello \n", "piper \n")

将文件加载到集合中但去掉空格和 \n 的最简单方法是什么？

谢谢

Answer 1

my_set = set(map(str.strip, open('filename.txt')))

Answer 2

仅删除右侧空格。

set(map(str.rstrip, open('filename.txt')))

Answer 3

with open("filename.txt") as f:
    s = set([line.rstrip('\n') for line in f])

Answer 4

with open("filename.txt") as f:
    mySet = map(str.rstrip, f)

如果你想在Python 2.5中使用它，你需要

from __future__ import with_statement

Answer 5

字符串的 strip() 方法删除两端的空格。

set(line.strip() for line in open('filename.txt'))

Answer 6

只需加载所有文件数据并将其拆分，它将处理每行一个单词或每行用空格分隔的多个单词，并且一次加载整个文件会更快，除非您的文件是 GB 的

words =  set(open('filename.txt').read().split())