Python:将单词从文件加载到

2024-04-25 12:28:31 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个简单的文本文件,有几千个单词,每一个都有自己的行,例如

aardvark
hello
piper

我使用以下代码将单词加载到一个集合中(我需要单词列表来测试成员资格,因此set是我选择的数据结构):

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

上面的代码生成一个集合,其中包含以下条目(每个单词后面跟着一个空格和新行字符:

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

将文件加载到一个集合中,但去掉空间并将其删除的最简单方法是什么?

谢谢


Tags: 代码txt数据结构hello列表my成员open
3条回答
my_set = set(map(str.strip, open('filename.txt')))

只需加载所有文件数据并将其拆分,它将处理每行一个字或每行用空格分隔的多个字,而且,如果文件不是以GBs为单位,则一次加载整个文件会更快

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

strip()方法从两端删除空白。

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

相关问题 更多 >