Question

摘要：

在Python项目中，我需要对具有相同内部索引和相邻外部索引的列表的每两个元素应用一个函数。输出将存储在新矩阵中。

我编写的代码可以工作，但是不够优雅，而pyflakes抱怨它。

如何清理此代码？

其他信息：

我正在编写的代码是解决数字难题的模块的一部分。

有一次我正在遍历类实例列表的列表。
它们代表运动场中成行的单元格。
我需要对两个垂直相邻的单元格应用一个函数，
并将其输出存储在新矩阵中。

在这里，配对中的哪个单元是最重要的，但是配对必须是有序的。

代码摘录：

def func(cell_matrix):
    out_matrix = []
    for y_pos, line in enumerate(cell_matrix):
        out_line = []
        if y_pos != 0:
            for x_pos, cell in enumerate(line):
                out_line.append(compare_func(prev_line[x_pos], cell)
            out_matrix.append(out_line)
        prev_line = line
    return out_matrix

pyflakes抱怨什么：

Line 7: pyflakes [E]: undefined name 'prev_line'
Line 9: pyflakes [E]: local variable 'prev_line' is assigned to but never used

Answer 1

使用变量之前，需要先声明变量名称：

def func(cell_matrix):
    out_matrix = []
    prev_line = cell_matrix[0]    # use 1st line as prev_line
    for line in cell_matrix[1:]:  # use 2nd to nth line, no y_pos used
                                  # in the following code so no need to enumerate
        out_line = []
            for x_pos, cell in enumerate(line):
                out_line.append(compare_func(prev_line[x_pos], cell) ) # missing )
            out_matrix.append(out_line)
        prev_line = line
    return out_matrix

Answer 2

我建议仅使用索引，这样可以节省prev_变量。

例如

def func(cell_matrix):                                                           
    out_matrix = []                                                              
    for y_pos in range(len(cell_matrix)):                                        
        out_line = []                                                            
        if y_pos != 0:                                                           
            for x_pos in range(len(cell_matrix[y_pos])):                         
                out_line.append(compare_func(cell_matrix[y_pos-1][x_pos],        
                                             cell_matrix[y_pos][x_pos]))         
            out_matrix.append(out_line)                                          
    return out_matrix

但是通过使用理解，它甚至可以进一步简化：

def func(cell_matrix):                                                           
    return [[compare_func(                                                       
                 cell_matrix[y_pos-1][x_pos], cell_matrix[y_pos][x_pos])         
             for x_pos in range(len(cell_matrix[y_pos]))]                        
            for y_pos in range(1, len(cell_matrix))]

编辑：顺便说一句，您得到的错误是pyflakes消息，代码运行良好。（一个人可能会说pyflakes无法正确解析代码）

Answer 3

将代码更改为：

def func(cell_matrix):
    out_matrix = []
    prev_line = []
    for y_pos, line in enumerate(cell_matrix):
        out_line = []
        if y_pos != 0:
            for x_pos, cell in enumerate(line):
                out_line.append(compare_func(prev_line[x_pos], cell)
            out_matrix.append(out_line)
        prev_line = line
    return out_matrix

您需要在for循环范围之外声明prev_line，以便它可以在每个循环中使用。

Answer 4

我最终将迭代移到了一个单独的函数上，因为我发现这种方式更易于阅读。

def subsequences(var, r=2):
    """Yield subsequences of var with length r"""
    # subsequences("ABCD") --> AB BC CD
    for index in range(len(var)-r+1):
        yield tuple(var[index+n] for n in range(r))

def func(cell_matrix):
    return [
        [
            compare_func(cell_a, cell_b)
            for cell_a, cell_b in zip(prev_line, line)
        ]
        for prev_line, line in subsequences(cell_matrix)
    ]

虽然可能只是个人喜好。

保持上一个项目在列表项中循环访问的一种优雅方法

4 个答案: