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

Question

I have a simple text file with several thousands of words, each in its own line, eg

aardvark
hello
piper

I use the following code to load the words into a set (I need the list of words to test membership, so set is the data structure I chose):

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

The above code produces a set with the following entries (each word is followed by a space and new-line character:

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

What's the simplest way to load the file into a set but get rid of the space and \\n?

Thanks

Answer 1

string的strip（）方法从两端移除空格。

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

Answer 2

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

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

Answer 3

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

Answer 4

仅删除右侧空格。

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

Answer 5

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

If you want to use this in Python 2.5, you need

from __future__ import with_statement

Answer 6

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