python - 包含最大值的字典

Question

我有这个 scipy csr_matrix:

  (0, 12114) 0.272571581001
  (0, 12001) 0.0598986479579
  (0, 11998) 0.137415042369
  (0, 11132) 0.0681428952502
  (0, 10412) 0.0681428952502
  (1, 10096) 0.0990242494495
  (1, 10085) 0.216197045661
  (1, 9105) 0.1362857905
  (1, 8925) 0.042670696769
  (1, 8660) 0.0598986479579
  (2, 6577) 0.119797295916
  (2, 6491) 0.0985172979468
  (3, 6178) 0.1362857905
  (3, 5286) 0.119797295916
  (3, 5147) 0.270246307076
  (3, 4466) 0.0540492614153
  (4, 3810) 0.0540492614153
  (4, 3773) 0.0495121247248

我想找到一种方法来创建(在本例中为 4 个)字典，其中每个字典包含每行的 2 个最大值..

例如，对于第 0 行，我的字典将是:

dict0 = {12114: '0.27257158100111998', 11998: '0.137415042369'}

对于第 1 行:

dict1 = {10085: '0.216197045661', 9105: '0.1362857905'}

Answer 1

由于csr_matrix没有sort()方法，所以先将需要的行转换为数组会很方便:

a = m[i,:].toarray().flatten()

获取已排序列的位置:

argsa = a.argsort()

最大值位于argsa的最后一列，因此获取两个最大值的列为:

argsa[-2:]

获取列、值对:

argsa[-2:], a[ argsa[-2:] ]

这可以在字典中转换:

dict( zip( argsa[-2:], a[ argsa[-2:] ] ) )

你的最终函数可以是这样的:

def get_from_m(m, i, numc=2):
    a = m[i,:].toarray().flatten()
    argsa = a.argsort()
    return dict( zip( argsa[-numc:], a[ argsa[-numc:] ] ) )