如何在重复索引时创建一个显示各个值之和的向量？

1 回答

• 0

  您可以创建一个字典，其中键是索引，值是总和：

  d = {
      'inds': [0, 3, 7, 3, 3, 5, 1],
      'vals': [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0]
  }
  
  result = {}
  for i, v in zip(d['inds'], d['vals']):
      if i not in result:
          result[i] = 0
      result[i] += v
  
  print(result)
  

  Output

  {0: 1.0, 1: 7.0, 3: 11.0, 5: 6.0, 7: 3.0}
  

  如果列表（向量）是强制性的，可以通过以下方式完成：

  result = [0]*(max(d['inds']) + 1)
  for i, v in zip(d['inds'], d['vals']):
      result[i] += v
  
  print(result)
  

  Output

  [1.0, 7.0, 0, 11.0, 0, 6.0, 0, 3.0]
  

  回复于 2024-04-25T16:51:33+08:00