Question

我有n x m矩阵，我想以编程方式找到N个或更多行，其中包含超过M个公共列的非零单元格。

例如。这是矩阵：

matrix([[ 0.,  0.,  1.,  1.,  1.,  0.,  1.,  0.],
        [ 1.,  0.,  1.,  0.,  1.,  1.,  0.,  1.],
        [ 1.,  0.,  0.,  0.,  0.,  1.,  0.,  0.],
        [ 0.,  0.,  0.,  0.,  1.,  0.,  1.,  0.],
        [ 0.,  1.,  0.,  1.,  1.,  0.,  1.,  0.],
        [ 0.,  1.,  0.,  1.,  0.,  0.,  0.,  0.]])

我正在寻找2行或更多行，其中包含2个或更多常见列中的非零单元格。有几种可能的结果，其中之一是：

row1: [ 1.,  0.,  1.,  0.,  1.,  1.,  0.,  1.],
row2: [ 1.,  0.,  0.,  0.,  0.,  1.,  0.,  0.],
      col1                     col5

是否可以找到解决此任务的所有行组合？

Answer 1

from itertools import combinations                   
matrix =[[ 0.,  0.,  1.,  1.,  1.,  0.,  1.,  0.],   
         [ 1.,  0.,  1.,  0.,  1.,  1.,  0.,  1.],   
         [ 1.,  0.,  0.,  0.,  0.,  1.,  0.,  0.],   
         [ 0.,  0.,  0.,  0.,  1.,  0.,  1.,  0.],   
         [ 0.,  1.,  0.,  1.,  1.,  0.,  1.,  0.],   
         [ 0.,  1.,  0.,  1.,  1.,  0.,  0.,  0.],   
         [ 0.,  1.,  0.,  1.,  1.,  0.,  0.,  1.]]   

m = 2                                                
n = 6                                                
req_rows = []                                        
ncm = [x for x in combinations(matrix,m)]            
for x in ncm:                                        
    if sum([1 for l in zip(*x) if not 0 in l])>=m:   
        print x

<强>输出

([0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 0.0, 0.0], [0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 0.0, 1.0])
([0.0, 0.0, 1.0, 1.0, 1.0, 0.0, 1.0, 0.0], [1.0, 0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 1.0])
([0.0, 0.0, 1.0, 1.0, 1.0, 0.0, 1.0, 0.0], [0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 1.0, 0.0])
([0.0, 0.0, 1.0, 1.0, 1.0, 0.0, 1.0, 0.0], [0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 1.0, 0.0])
([0.0, 0.0, 1.0, 1.0, 1.0, 0.0, 1.0, 0.0], [0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 0.0, 0.0])
([0.0, 0.0, 1.0, 1.0, 1.0, 0.0, 1.0, 0.0], [0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 0.0, 1.0])
([1.0, 0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 1.0], [1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0])
([1.0, 0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 1.0], [0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 0.0, 1.0])
([0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 1.0, 0.0], [0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 1.0, 0.0])
([0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 1.0, 0.0], [0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 0.0, 0.0])
([0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 1.0, 0.0], [0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 0.0, 1.0])
([0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 0.0, 0.0], [0.0, 1.0, 0.0, 1.0, 1.0, 0.0, 0.0, 1.0])

Answer 2

from pprint import pprint
from itertools import combinations

def solve(lst, m):

    col, n = {}, len(lst)
    for i, x in enumerate(lst):
        col[i] = [j for j, y in enumerate(x) if y]

    for s in xrange(n, m-1, -1):
        for c in combinations(xrange(n), s):
            values  = set(col[c[0]]).intersection(*(col[k] for k in c[1:]))
            if len(values) >= m:
                yield [lst[k] for k in c]

for x in solve(matrix, 2):
    pprint(x)

<强>输出：

[[0, 0, 1, 1, 1, 0, 1, 0],
 [0, 0, 0, 0, 1, 0, 1, 0],
 [0, 1, 0, 1, 1, 0, 1, 0]]
[[0, 0, 1, 1, 1, 0, 1, 0], [1, 0, 1, 0, 1, 1, 0, 1]]
[[0, 0, 1, 1, 1, 0, 1, 0], [0, 0, 0, 0, 1, 0, 1, 0]]
[[0, 0, 1, 1, 1, 0, 1, 0], [0, 1, 0, 1, 1, 0, 1, 0]]
[[1, 0, 1, 0, 1, 1, 0, 1], [1, 0, 0, 0, 0, 1, 0, 0]]
[[0, 0, 0, 0, 1, 0, 1, 0], [0, 1, 0, 1, 1, 0, 1, 0]]
[[0, 1, 0, 1, 1, 0, 1, 0], [0, 1, 0, 1, 0, 0, 0, 0]]

Python：如何在大于M个公共列中找到大于N行且非零单元格

2 个答案: