如何减少if语句并缩短代码?

时间:2019-10-29 16:59:03

标签: python if-statement refactoring

我的代码有效,但是,弹出一条错误消息,指出语句太多。

我不知道如何减少语句,因为每个语句在函数中都有其用途。

def is_sink(elevation_map: List[List[int]], cell: List[int]) -> bool:
    """Return True if and only if cell exists in the elevation map
    elevation_map and cell is a sink.

    Precondition: elevation_map is a valid elevation map.
                  cell is a 2-element list.

    >>> is_sink(THREE_BY_THREE, [0, 5])
    False
    >>> is_sink(THREE_BY_THREE, [0, 2])
    True
    >>> is_sink(THREE_BY_THREE, [1, 1])
    False
    >>> is_sink(FOUR_BY_FOUR, [2, 3])
    True
    >>> is_sink(FOUR_BY_FOUR, [3, 2])
    True
    >>> is_sink(FOUR_BY_FOUR, [1, 3])
    False
    """    
    x = cell[0]
    y = cell[1]
    x_initial = 0
    x_final = 0
    y_initial = 0
    y_final = 0

    if x > len(elevation_map): 
        return False
    elif y > len(elevation_map[0]): 
        return False 
    if x - 1 < 0:
        x_initial = x
    else:
        x_initial = x - 1
    if x + 1 >= x:
        x_final = x
    else:
        x_final = x + 1
    if y - 1 < 0:
        y_initial = y
    else:
        y_initial = y - 1
    if y + 1 >= y:
        y_final = y
    else:
        y_final = y + 1
    validsink = True
    for i in range(x_initial, x_final + 1):
        for j in range(y_initial, y_final + 1):
            if elevation_map[i][j] < elevation_map[x][y]:
                validsink = False
    return validsink 

R0915(太多陈述)出现次数:1。   [第142行]语句过多(24/20)

1 个答案:

答案 0 :(得分:0)

这看起来像是py-lint警告,表明您的代码试图在一个例程中做太多事情。并不是真正的错误,但是它确实指向了不理想的编码实践。

您可以减少其中的一些内容。

if x - 1 < 0:
    x_initial = x
else:
    x_initial = x - 1

可以成为

x_initial = x if x - 1 < 0 else x - 1

,对于x_final,y_initial和y_final等。