垂直或水平计算列表列表中“连续”的值

Question

我正在编写一个函数来计算列表列表中特定值的出现次数连续水平或垂直。然后它只需要返回它发生了多少次的值。这是一个例子

lst=[['.','.','.','e'],
     ['A','A','.','e'],
     ['.','.','.','e'],
     ['.','X','X','X'],
     ['.','.','.','.'],
     ['.','.','.','e']]

对于这个列表列表，函数应该为 e 返回 3，因为它连续出现 3 次，为 A 为 2，为 X 为 3。谢谢您的时间

到目前为止我的代码：

def length_of_row(symbol,lot):
    count = 0
    for sublist in lot:
        for x in sublist:
            if x == symbol:
                count += 1
                continue
            else:
                continue
                return count

Answer 1

如果您不介意稍作改动，可以尝试以下方法：

from functools import reduce
from itertools import takewhile


def length_of_row(symbol, lot):
    if symbol not in reduce(lambda x,y: x+y, lot):
        return 0
    elif symbol in lot[0]:
        good_lot =  map(lambda y: y.count(symbol),takewhile(lambda x: symbol in x, lot))
        return sum(good_lot)
    else:
        return length_of_row(symbol, lot[1:])

这结合了递归和 python 强大的itertools 方法之一 (takewhile)。这个想法是计算符号的数量，直到您点击一个不包含该符号的列表。此外，它会尝试确保仅当符号在列表列表中时才计算符号的出现次数。

使用它：

lst = [['.', '.', '.', 'e'],
 ['A', 'A', '.', 'e'],
 ['.', '.', '.', 'e'],
 ['.', 'X', 'X', 'X'],
 ['.', '.', '.', '.'],
 ['.', '.', '.', 'e']]

print(length_of_row('e', lst))
print(length_of_row('X', lst))
print(length_of_row('A', lst))
print(length_of_row('f', lst))

#3
#3
#2
#0

如您所见，如果符号不存在，则返回0。

编辑：

如果您不想从itertools 导入takewhile 函数，您可以使用文档中提供的近似定义。但请记住，它不如itertools 方法优化：

def takewhile(predicate, iterable):
    for x in iterable:
        if predicate(x):
            yield x
        else:
            break

此外，如果您使用的是python2，您应该可以直接使用reduce。但是，您可以定义一个函数来将列表列表缩减为一个列表，如下所示：

def reduce_l_of_l(lst_of_lst):
    out_lst = []
    for lst in lst_of_lst:
        out_lst += lst
    return out_lst

不要使用reduce，而是在定义后将其替换为reduce_l_of_l。

我希望这会有所帮助。

Answer 2

这实际上是一个很难用基本原理解决的问题，如果您刚开始学习编程，这将尤其困难。这是一个简洁但更高级的解决方案：

result = {}
for grid in [lst, zip(*lst)]:
    for row in grid:
        for key, group in itertools.groupby(row):
            result[key] = max(len(list(group)), result.get(key, 0))

那么result就是：

{'A': 2, 'X': 3, 'e': 3, '.': 4}