在python中读取文件后返回单词列表

我有一个名为test.txt的文本文件.我想阅读它并从文件中返回所有单词列表(删除换行符).

这是我目前的代码:

def read_words(test.txt):
    open_file = open(words_file, 'r')
    words_list =[]
    contents = open_file.readlines()
    for i in range(len(contents)):
         words_list.append(contents[i].strip('\n'))
    return words_list    
    open_file.close()

运行此代码会生成以下列表:

['hello there how is everything ', 'thank you all', 'again', 'thanks a lot']

我希望列表看起来像这样:

['hello','there','how','is','everything','thank','you','all','again','thanks','a','lot']
用以下内容替换for循环中的words_list.append(…)行:

words_list.extend(contents[i].split())

这将在空格字符上拆分每一行,然后将结果列表的每个元素添加到words_list.

或者作为将整个函数重写为列表推导的替代方法:

def read_words(words_file):
    return [word for line in open(words_file, 'r') for word in line.split()]
相关文章
相关标签/搜索