在Python中搜索四元组中的元组

Question

我有一组四元组('myTup')和一个元组('tupleToSearch')。我需要将这个元组的所有实例搜索到每个四元组中。“tupleToSearch”将针对每个四元组的前两个元素进行比较，如果匹配，则将报告匹配的四元组的最后两个元素。

我使用下面的代码来做同样的事情。

myTup = set([('0994900', '50.2297', 'name1', '<4'),
             ('2176041', '24.2880', 'name2', 'POSITIVE'), 
             ('2240663', '51.2510', 'name3', '25.0'), 
             ('2240663', '51.2510', 'name4', '29.0'), 
             ('2240663', '51.2560', 'name4', '29.0')])

tupToSearch = ('2240663', '51.2510')


[(x[2],x[3]) for x in myTup if tupToSearch == (x[0],x[1])]

我需要扩展这段代码，以便它不是精确搜索，而是对范围执行比较。

例如，给定tupleToSearch = ('2240663', '51.2510')，我希望找到第二个元素是>= 51.2510但小于(51.2510 +偏移量)的那些四元组。这里'offset‘是一个常量。

上例中的正确答案将报告最后三个四元组(每个元素中只有最后两个元素)。

如何将第二个元素转换为数值以进行正确的比较。

此外，我需要一个有效的方法来做到这一点，因为我需要重复这个步骤几乎十亿次。

Answer 1

过滤掉你想要的元素，然后拉出每个元素的块来创建你的结果。

[(el[2], float(el[3])) for el in itertools.ifilter(lambda oel: oel[0] == search_tup[0] and float(search_tup[1]) <= float(oel[1]) <= float(search_tup[1]) + offset, my_tup)]

或者，使用生成器一次完成所有操作(请注意，生成器与list comp不同，它将优雅地处理el[3]不能转换为浮点型的情况)。

def filtered_data(input, search_target, offset):
    key = search_target[0]
    value = float(search_target[1])
    for entry in input:
        entry_value = float(entry[1])
        if entry[0] == key and entry_value <= value <= entry_value + offset:
            try:
                result_value = float(entry[3])
            except ValueError:
                result_value = entry[3]
            yield (entry[2], result_value)

[filtered_data(my_tup, search_tup, .019)]

请注意，由于您在此处使用的是浮点值，因此在将浮点十进制值转换为浮点二进制值时，您的过滤过程将始终受到错误误差的影响。因此，您可能还需要使用Decimal或在基值中构建一个偏移量。

Answer 2

好了，这里有一个float function，可以用来写下面的代码：

low_target = float(tupToSearch[1])
high_target = low_target + constant
[(x[2],x[3]) for x in myTup if low_target<=float(x[1])<=high_target]