如何干净地遍历两个文件并行在Python

Question

我经常写这样的代码：

lines = open('wordprob.txt','r').readlines() 
words = open('StdWord.txt','r').readlines() 
i = 0 
for line in lines: 
    v = [eval(s) for s in line.split()] 
    if v[0] > v[1]: 
     print words[i].strip(), 
    i += 1 

是否有可能避免使用变量i和使程序更短？

谢谢。

Answer 1

你可以尝试使用枚举，

http://docs.python.org/tutorial/datastructures.html#looping-techniques

lines = open('wordprob.txt','r').readlines() 
words = open('StdWord.txt','r').readlines() 
for i,line in enumerate(lines): 
     v = [eval(s) for s in line.split()] 
     if v[0] > v[1]: 
       print words[i].strip() 

Answer 2

在enumerate看看：

>>> for i, season in enumerate(['Spring', 'Summer', 'Fall', 'Winter']): 
...  print i, season 
0 Spring 
1 Summer 
2 Fall 
3 Winter 

Answer 3

一般来说枚举是一个很好的解决方案。在这种情况下，你可以这样做：

lines = open('wordprob.txt','r').readlines() 
words = open('StdWord.txt','r').readlines() 
for word, line in zip(words, lines): 
    v = [eval(s) for s in line.split()] 
    if v[0] > v[1]: 
      print word.strip(), 

Answer 4

它看起来像你不在乎什么i值。您只是将它用作配对lines和words的方法。因此，我建议您一次读一行，并同时读一个字。然后他们会匹配。

此外，当您使用.readlines()时，您将所有输入一次读入内存。对于大量输入，这会很慢。对于这个简单的代码，一次只需要一行代码。由open()返回的文件对象可以充当一次返回一行的迭代器。

如果可以，应该避免使用eval()。在一个简单的练习中，您知道输入的数据是什么，这非常安全，但是如果您从外部获取数据，使用eval()可能会使您的计算机受到攻击。有关更多信息，请参见this page。我将编写我的示例代码，假设您使用eval()将文本转换为float值。 float()也可以使用整数字符串值：float('3')将返回3.0。

此外，看起来输入行只能有两个值。如果一行有额外的值，你的代码将不会检测到这种情况。我们可以更改代码以明确地从分割线中解压缩两个值，然后如果有两个以上的值，Python将引发异常。另外，代码会稍微好一些。

因此，这里是我建议的这个例子的改写：

lines = open('wordprob.txt','rt') 
words = open('StdWord.txt','rt') 

for line in lines: 
    word = words.next().strip() # in Python 3: word = next(words).strip() 
    a, b = [float(s) for s in line.split()] 
    if a > b: 
     print word, # in Python 3: print(word + ' ', end='') 

编辑：这里是同一个解决方案，但使用izip()。

import itertools 
lines = open('wordprob.txt','rt') 
words = open('StdWord.txt','rt') 

# in Python 3, just use zip() instead of izip() 
for line, word in itertools.izip(lines, words): 
    word = word.strip() 
    a, b = [float(s) for s in line.split()] 
    if a > b: 
     print word, # in Python 3: print(word + ' ', end='') 

在Python 3，内置zip()返回迭代器，所以你可以使用，而不是需要import itertools。

编辑： 最好的做法是使用with语句来确保文件正确关闭，无论如何。在Python的最新版本中，您可以使用多个语句，我将在我的解决方案中执行此操作。另外，我们可以像解开一个列表一样简单地解压缩生成器表达式，所以我改变了将a, b设置为使用生成器表达式的行;这应该稍微快一点。除非我们打算使用它，否则我们不需要去除word。把更改放在一起得到：

from itertools import izip 

with open('wordprob.txt','rt') as lines, open('StdWord.txt','rt') as words: 
    # in Python 3, just use zip() instead of izip() 
    for line, word in izip(lines, words): 
     a, b = (float(s) for s in line.split()) 
     if a > b: 
      print word.strip(), # in Python 3: print(word.strip() + ' ', end='')