如何从文本文件中仅添加非零内容？

Question

我在文本文件中有以下类型的数据。

15  1  
23  0  
39 -1  
71 -1  
79  1  
95  1  
127 -2  
151  2  
183  1  
191 -1  
239  0  
247  3  

我想从文本文件创建一个二维列表，如下所示。我可以使用下面给出的代码来做到这一点，结果如下

[[15, 1.36896146582243],  
[23, 0.000000000000000],  
[39, 0.848993860380692],  
[71, 0.629227476540724],  
[79, 0.596517662620081],  
[95, 0.543970127117099],  
[127, 1.88189324753006],  
[151, 1.72587115688942],  
[183, 0.391932527534896],  
[191, 0.383636720228727]]  

但是我不想要所有条目，我只想要源文本文件第二列中非零条目的条目。例如我不想要条目

23  0
239 0

如何将条件语句添加到我的代码中。

with open("path.text") as file:
    R = [[int(x) for x in line.split()] for line in file]

Answer 1

pythonic 的解决方案是在列表推导中添加一个 if 语句

使用 open("path.text") 作为文件： R = [[int(x) for x in line.split()] for line in file if line[1] != '0']

Answer 2

没有必要把它硬塞到一个列表理解表达式中——在你的情况下它不会更快：

result = []
with open("path.text", "r") as f:
    for line in f:
        line = line.split()
        if len(line) < 2:  # just to make sure we have both columns
            continue
        column_2 = float(line[1])  # since, it appears, you have floats
        if column_2:
            result.append([int(line[0]), column_2])  # turn column 1 to int, too

更新 - 基准时间 - 如果您将函数定义为彼此紧密匹配（因此没有上述浮动处理或验证）：

def f1():
    with open("path.text", "r") as f:
        return [[int(x) for x in line.split()] for line in f if '0' not in line.split()]

def f2():
    result = []
    with open("path.text", "r") as f:
        for line in f:
            line = line.split()
            if line[1] != '0':
                result.append([int(line[0]), int(line[1])])
    return result

path.text 包含与 OP 和 assert f1() == f2() 传递相同的数据，这是我系统上的一些结果：

 Python 3.5.1, 64-bit
 f1() - 100,000 loops: 10.834s
 f2() - 100,000 loops: 9.9601s

 Python 2.7.11 64-bit
 f1() - 100,000 loops: 6.9243s
 f2() - 100,000 loops: 6.4012s

其中大部分实际上是 I/O，相对而言，处理上的差异实际上要大得多。