python - 查找Python列表中的所有矩形

Question

我有一个 python 列表:

[
[1, 1, 2],
[1, 1, 2],
[7, 4, 5],
[5, 3, 7],
]

我需要找到这个二维数组中满足此条件的所有矩形(有多少个矩形):
1. 一个矩形内的所有数字必须相同
2. 该矩形中的数字不能位于数组中的其他位置
3. 并不是真正的标准，而是一些额外的信息:
矩形尺寸可以是 1x1、2x1、2x2、3x1、3x2、3x3、3x4 等。

在此示例中，有 5 个数字矩形:

1, 2, 3, 4

数字 3、1、8 不符合条件，因为:

5 - 违反规则 2
7 - 违反规则 2

我尝试过类似查找最近的元素来比较它们，如果它们匹配则向下一行等等，但我无法做到这一点，所以我希望有人可以提供帮助。谢谢。

Answer 1

一种方法是对于每个唯一值，在该值的所有实例上拟合一个矩形。如果该矩形内的所有值都匹配，则您已满足标准。这里使用 numpy 在代码中实现，打印符合条件的值:

import numpy as np

arr = np.array([
    [3, 2, 2, 1, 4],
    [3, 2, 2, 7, 4],
    [8, 2, 2, 1, 3],
    [8, 8, 9, 9, 9],
    [8, 8, 1, 5, 1]
])
for n in np.unique(arr):
    y, x = np.where(arr == n)
    if (arr[y.min():y.max()+1, x.min():x.max()+1] == n).all():
        print(n)

---

更新

不太漂亮，但这样的东西不需要 numpy:

lst = [
    [3, 2, 2, 1, 4],
    [3, 2, 2, 7, 4],
    [8, 2, 2, 1, 3],
    [8, 8, 9, 9, 9],
    [8, 8, 1, 5, 1]
]

for val in set([x for sub in lst for x in sub ]):
    y = [n for n, sub in enumerate(lst) if val in sub]
    if any(y):
        x = []
        for sub in [lst[n] for n in y]:
            x.extend([m for m, v in enumerate(sub) if v == val])

        rect = [i for sub in lst[min(y):max(y)+1] for i in sub[min(x):max(x)+1]]
        if all([i == val for i in rect]):
            print(val)