python - 将文件读入列表并去除换行符

Question

我在将文件读入列表时遇到问题，当我这样做时，它只会从整个文件中创建一个项目，而不是将每个元素读入其自己的字段中。我正在使用\n作为剥离的东西，但我无法让它正常工作。

temp = open('drugs')
drugs = [temp.read().strip("\n")]
temp.close

结果：

['40 Stimpak\n53 Mentats\n87 Buffout\n109 Rad-X\n125 Booze\n260 Jet Antidote\n311 Roentgen Rum\n424 Monument Chunk\n480 Bonus +1 Agility\n525 Hypo \n48 RadAway\n71 Fruit\n103 Iguana-on-a-stick\n110 Psycho\n144 Super Stimpak\n273 Healing Powder\n334 Poison\n469 Rot Gut\n481 Bonus +1 Intelligence \n49 Antidote\n81 Iguana-on-a-stick\n106 Nuka-Cola\n124 Beer\n259 Jet\n310 Gamma Gulp Beer\n378 Cookie\n473 Mutated Toe\n482 Bonus +1 Strength ']
drugs.strip('\n')

Traceback (most recent call last):
   File "seek", line 18, in <module>
     print drugs.strip('\n')
AttributeError: 'list' object has no attribute 'strip'

Answer 1

file.read()读取整个文件的内容，除非您指定最大长度。你必须是什么意思.readlines()。但是您可以使用列表理解更加地道：

with open('drugs') as temp_file:
  drugs = [line.rstrip('\n') for line in temp_file]

该with语句将负责关闭文件。

Answer 2

如果您可以将整个文件的内容读入内存，您也可以使用str.splitlines()

with open('your_file.txt') as f:
    lines = f.read().splitlines()

splitlines()类似于split('\n')但如果您的文件以换行符结尾，split('\n')则将在最后返回一个空字符串，而splitlines()以您想要的方式处理这种情况。

Answer 3

这将条带直接合并到for语句中。

with open('drugs', 'r') as f:
  for line in map(lambda line: line.rstrip('\n'), f):
    print line

或者，如果您知道一行文本前后不需要任何空格，则可以使用它。

import string

with open('drugs', 'r') as f:
  for line in map(string.strip, f):
    print line