创建一个稀疏的coo_matrix并在两个for循环结构中进行更新

Question

在外部for循环中，我将k1作为x坐标，在内部循环中，我将k2作为y坐标。 循环基本上是用于其他目的，即创建临时字典（用于其他目的）。 因此，我无法删除循环结构。 我正在尝试创建一个coo_matrix，其坐标为k1 ， k2 ，其值是在内部循环中计算的。

这是我的简短代码：

for k1,v1 in s.iteritems():
    xdict[k1] = None
    for k2,v2 in s.iteritems():
        tempdict = {}
        if k2 in xdict.keys():
            continue
        if k1 == k2:
            continue
        tempdict[k1] = v1
        tempdict[k2] = v2
        r = reduce(set.intersection, (set(val) for val in tempdict.values()))
        new_kee = 'C(' + k1 + ',' + k2 + ')'
        ydict[new_kee] = len(r)

输入“ s”是具有以下示例键和值的字典：

216 [234]  
1305 [234, 234, 298, 429]  
2961 [241, 264]  
211 [143]  
....  

内部循环计算两个记录中每个记录的交集而无需重复。 在我的新字典中，相交结果的输出将存储为：

c(216,1305) -> 1  
c(216,2961) -> 0  
c(216,211) -> 0  
c(1305,2961) -> 0  
c(1305,211) -> 0  
c(2961,211) -> 0  
...  

在这一点上，我想将值存储在稀疏的coo_matrix中，其坐标如上计算，例如：

• 对于坐标（216,1305），将存储值1
• 对于坐标（216,2961），将存储值0
  等等....

我需要在坐标(k1,k2)中创建值len(r)的稀疏coo_matrix的帮助。 我不确定是否必须在所有具有固定大小的循环结构之前创建矩阵，然后在循环内或以其他任何方式更新。

有人可以帮我解决这个问题吗？

Answer 1

这就是我要做的。 定义两个功能：

def new_matrix(n,m):
    """Create a new matrix of dimensions 'n' by 'm'."""
    return [[0 for col in range(m)] for row in range(n)]

def update_matrix(matrix,row,col,val):
    """Update a matrix at row,col position with the value 'val'."""
    matrix[row][col] = val

如下更新代码：

#You need to know the size ahead of time
#Replace row, col with appropriate dimensions
matrix = new_matrix(row,col)

for k1,v1 in s.iteritems():
    xdict[k1] = None
    for k2,v2 in s.iteritems():
        tempdict = {}
        if k2 in xdict.keys():
            continue
        if k1 == k2:
            continue
        tempdict[k1] = v1
        tempdict[k2] = v2
        r = reduce(set.intersection, (set(val) for val in tempdict.values()))
        new_kee = 'C(' + k1 + ',' + k2 + ')'
        ydict[new_kee] = len(r)

        #Update your matrix here:
        update_matrix(matrix,k1,k2,len(r))