在具有容差范围的元组中对连续数字进行分组

时间:2015-03-20 06:09:29

标签: python list range continuous

如果我有一组元组:

locSet = [(62.5, 121.0), (62.50000762939453, 121.00001525878906), (63.0, 121.0),(63.000003814697266, 121.00001525878906), (144.0, 41.5)]

我想将它们分组,公差范围为+/- 3。

aFunc(locSet)

返回

[(62.5, 121.0), (144.0, 41.5)]

我见过Identify groups of continuous numbers in a list,但那是连续的整数。

1 个答案:

答案 0 :(得分:0)

如果我理解得很好,那么您正在搜索其值在容差范围内的绝对量不同的元组:[0,1,2,3]

假设这样,我的解决方案返回一个列表列表,其中每个内部列表都包含满足条件的元组。

def aFunc(locSet):
  # Sort the list.
  locSet = sorted(locSet,key=lambda x: x[0]+x[1])

  toleranceRange = 3
  resultLst = []
  for i in range(len(locSet)):
      sum1 = locSet[i][0] + locSet[i][1]
      tempLst = [locSet[i]]
      for j in range(i+1,len(locSet)): 
          sum2 = locSet[j][0] + locSet[j][1]
          if (abs(sum1-sum2) in range(toleranceRange+1)):
              tempLst.append(locSet[j])

      if (len(tempLst) > 1):
          for lst in resultLst:
              if (list(set(tempLst) - set(lst)) == []):
                  # This solution is part of a previous solution.
                  # Doesn't include it.
                  break
          else:
              # Valid solution.
              resultLst.append(tempLst)

  return resultLst

这里有两个使用示例:

locSet1 = [(62.5, 121.0), (62.50000762939453, 121.00001525878906), (63.0, 121.0),(63.000003814697266, 121.00001525878906), (144.0, 41.5)]
locSet2 = [(10, 20), (12, 20), (13, 20), (14, 20)]

print aFunc(locSet1)
[[(62.5, 121.0), (144.0, 41.5)]]

print aFunc(locSet2)
[[(10, 20), (12, 20), (13, 20)], [(12, 20), (13, 20), (14, 20)]]

我希望得到帮助。