时间戳

首页 > 解决方案 > 合并包含相同值的数组

python - 合并包含相同值的数组

问题描述

我需要从相关系数矩阵中得到一个高相关组，保留其中一个并排除另一个。但我不知道如何优雅高效地做到这一点。

这是一个类似的答案，但希望它会使用类似向量的矩阵来完成。： 如果数组包含一个或多个相同的值，则合并数组

例如：

a = np.array([[1,0,0,0,0,1],
              [0,1,0,1,0,0],
              [0,0,1,0,1,1],
              [0,1,0,1,0,0],
              [0,0,1,0,1,0],              
              [1,0,1,0,0,1]])

对角线：

(0,0),(1,1),(2,2)...(5,5)

其他：

(0,5),(1,3),(2,4),(2,5)

这三对因为相互包含合并成一组：</p> 

(0,2,4,5) = (0,5),(2,4),(2,5)

所以最终我需要输出：（我将使用结果来索引其他数据，因此决定保留每组中最大的索引）

out = [(0,2,4,5),(1,3)]

我认为最简单的方法是采用嵌套循环并多次遍历所有元素。想有更简洁高效的实现方式，谢谢

这是一个循环实现，不好意思写得很辛苦：</p> 

a = np.array([[1,0,0,0,0,1],
              [0,1,0,1,0,0],
              [0,0,1,0,1,1],
              [0,1,0,1,0,0],
              [0,0,1,0,1,0],              
              [1,0,1,0,0,1]])

a[np.tril_indices(6, -1)]= 0     
a[np.diag_indices(6)]    = 0     
g = list(np.c_[np.where(a)])

p = {}; index = 1
while len(g)>0:
    x = g.pop(0)
    if not p:
        p[index] = list(x)
        for i,l in enumerate(g):
            if np.in1d(l,x[0]).any()|np.in1d(l,x[1]).any():
                n = list(g.pop(i))
                p[index].extend(n)
    else:
        T = False
        for key,v in p.items():
            if np.in1d(v,x[0]).any()|np.in1d(v,x[1]).any():
                v.extend(list(x))
                T = True
        if T==False:
            index += 1; p[index] = list(x)
            for i,l in enumerate(g):
                if np.in1d(l,x[0]).any()|np.in1d(l,x[1]).any():
                    n = list(g.pop(i))
                    p[index].extend(n)

for key,v in p.items():
    print key,np.unique(v)

出去：

1 [0 2 4 5]
2 [1 3]

标签： pythonnumpy

解决方案

使用此答案可以解决合并/合并具有共同极值的对的核心问题。

因此，上面的代码可以重写为：

a = np.array([[1,0,0,0,0,1],
              [0,1,0,1,0,0],
              [0,0,1,0,1,1],
              [0,1,0,1,0,0],
              [0,0,1,0,1,0],              
              [1,0,1,0,0,1]])

a[np.tril_indices(6, -1)]= 0     
a[np.diag_indices(6)]    = 0     
g = np.c_[np.where(a)].tolist()

def consolidate(items):
    items = [set(item.copy()) for item in items]
    for i, x in enumerate(items):
        for j, y in enumerate(items[i + 1:]):
            if x & y:
                items[i + j + 1] = x | y
                items[i] = None
    return [sorted(x) for x in items if x]

p = {i + 1: x for i, x in enumerate(sorted(consolidate(g)))}

推荐阅读