python比较dictionary的值并丢弃它们

Question

我有一个字典，每个键有一个键和多个值：

d={'POU': ['GL', '1', '999', '4646']
   'TSA': ['LA', '2', '888', '4545']
   'RAS': ['NA', '5', '565', '1316']
   'TSO': ['RA', '7', '575', '1376']}

每个键的第二个值表示距离（1、2、5、7），单位为km。按键按距离排序。你知道吗

我要比较下列键的距离，并丢弃彼此靠近且距离等于或小于1km的键。 因此，最终的词典将是：

d={'POU': ['GL', '1', '999', '4646']
   'RAS': ['NA', '5', '565', '1316']
   'TSO': ['RA', '7', '575', '1376']}   

如何做到这一点？你知道吗

到目前为止，我所做的只是根据距离限制按键，例如：

if (float(d[key][1])<10):

但我无法将每个键的值与下一个键的值进行比较。我不喜欢Python，这两天我都被它弄疯了。你知道吗

Answer 1

您可以遍历键并在条件匹配时移除它们。你知道吗

# save keys in a list
vals = list(d.keys())

## count keys
dict_len = len(vals) 

# save keys to be removed
to_rmv = [] 

# iterate through keys
for i in range(dict_len):
    for j in range(i+1, dict_len):
        p = int(d[vals[j]][1])- int(d[vals[i]][1]) ## second index is distance
        if p <= 1:
            to_rmv.append(vals[j])            

## remove keys
for k in to_rmv:
    del d[k]

## see output
print(d)

{'POU': ['GL', '1', '999', '4646'],
 'RAS': ['NA', '5', '565', '1316'],
 'TSO': ['RA', '7', '575', '1376']}

Answer 2

keys_to_remove = []
for k1, v1 in d.items():
    for k2, v2 in d.items():
        if k1 == k2:
            continue
        dist1 = float(v1[1])
        dist2 = float(v2[1])
        if abs(dist1 - dist2) <= 1:
            keys_to_remove.append(k2)
result = d.copy()
for k in keys_to_remove:
    result.pop(k)
print(result)