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)