如何基于前两个元素检查Python列表中的行

嘿，针对你给出的dfdist_list列表，我来分享几种基于前两个元素检查行的常见实现方式，你可以根据实际需求调整：

先回顾你的示例数据 #

先把你给出的前5行数据整理成更易读的格式，方便后续测试：

dfdist_list = [
    [10255, 10256, (-89.572, 41.235), (-89.025, 41.431000000000004), 37.963909],
    [10255, 937, (-89.572, 41.235), (-101.209, 36.184), 11622.426262],
    [10255, 3392, (-89.572, 41.235), (-101.221, 36.008), 11621.601622],
    [10255, 306, (-89.572, 41.235), (-101.53399999999999, 36.241), 11599.87664],
    [10255, 7313, (-89.572, 41.235), (-103.331, 35.109), 11475.272909]
]

1. 检查是否存在重复的前两个元素组合 #

如果需要找出列表中前两个元素重复的行（比如避免ID对重复），可以用集合记录已出现的组合：

seen_pairs = set()
duplicate_rows = []

for row in dfdist_list:
    pair = (row[0], row[1])
    if pair in seen_pairs:
        duplicate_rows.append(row)
    else:
        seen_pairs.add(pair)

if duplicate_rows:
    print("找到重复的行：")
    for row in duplicate_rows:
        print(row)
else:
    print("没有重复的前两个元素组合")

2. 查找前两个元素匹配特定值的行 #

比如要精准定位前两个元素是(10255, 937)的行，用列表推导式就能快速实现：

target_pair = (10255, 937)
matching_rows = [row for row in dfdist_list if (row[0], row[1]) == target_pair]

print(f"匹配{target_pair}的行：")
for row in matching_rows:
    print(row)

3. 验证前两个元素是否符合自定义规则 #

比如你想检查第一个元素是否大于10000，第二个元素是否是三位数，可以这么写：

invalid_rows = []
for row in dfdist_list:
    first_id = row[0]
    second_id = row[1]
    # 这里可以替换成你实际需要的规则
    if not (first_id > 10000 and 100 <= second_id <= 999):
        invalid_rows.append(row)

if invalid_rows:
    print("不符合规则的行：")
    for row in invalid_rows:
        print(row)
else:
    print("所有行的前两个元素都符合规则")

4. 统计每个前两个元素组合的出现频次 #

如果需要统计不同ID对的出现次数，用collections.defaultdict会很方便：

from collections import defaultdict

pair_count = defaultdict(int)
for row in dfdist_list:
    pair = (row[0], row[1])
    pair_count[pair] += 1

print("前两个元素组合的出现次数：")
for pair, count in pair_count.items():
    print(f"{pair}: {count}次")

这些都是日常开发中比较常用的场景，你可以根据自己实际的检查逻辑修改代码细节～

内容的提问来源于stack exchange，提问作者Emily Beth