Python字典列表筛选异常：条件未生效原因及正确实现方法

原代码及输出 #

用户运行的Python代码：

positions = {
    'IBM': [
        {'ticker': 'IBM', 'start_date': '20240415', 'end_date': 99999999},
        {'ticker': 'IBM', 'start_date': '20240416', 'end_date': 00000000},
        {'ticker': 'IBM', 'start_date': '20240417', 'end_date': 99999999}],
    'MRK': [
        {'ticker': 'mmm', 'start_date': '20240415', 'end_date': 99999999},
        {'ticker': 'mmm', 'start_date': '20240416', 'end_date': 00000000},
        {'ticker': 'mmm', 'start_date': '20240417', 'end_date': 99999999}]} 

new_pos = [position for position in positions.values() 
           if position[0].get('end_date') == 99999999] 

print(new_pos)

得到的输出：

[[{'ticker': 'IBM', 'start_date': '20240415', 'end_date': 99999999}, 
  {'ticker': 'IBM', 'start_date': '20240416', 'end_date': 0}, 
  {'ticker': 'IBM', 'start_date': '20240417', 'end_date': 99999999}], 
 [{'ticker': 'mmm', 'start_date': '20240415', 'end_date': 99999999}, 
  {'ticker': 'mmm', 'start_date': '20240416', 'end_date': 0}, 
  {'ticker': 'mmm', 'start_date': '20240417', 'end_date': 99999999}]]

用户疑问：为何if条件未生效，返回了所有记录而非仅第一个符合条件的字典？猜测是因每个列表的第一个字典end_date为99999999时就返回整个列表，如何仅获取end_date等于99999999的字典？

问题原因 #

你的猜测完全正确。原列表推导式的逻辑是：遍历positions.values()中的每个完整列表（比如IBM对应的3个字典组成的列表），判断条件是「该列表的第一个元素的end_date是否为99999999」。只要满足这个条件，就把整个列表加入new_pos，而非筛选列表内的单个字典。因为两个公司的列表第一个元素都符合条件，所以最终返回了两个完整的列表。

另外注意：代码里的00000000是整数字面量，Python会自动解析为0，这就是输出里end_date显示为0的原因，如果需要保留8位数字格式，应该写成字符串'00000000'。

解决方法 #

要筛选出所有end_date等于99999999的单个字典，需要嵌套遍历：先遍历每个公司的子列表，再遍历子列表里的每个字典，判断并收集符合条件的项。

方法1：嵌套列表推导式 #

这是最简洁的写法：

new_pos = [item for sublist in positions.values() for item in sublist if item['end_date'] == 99999999]

运行后得到的输出：

[{'ticker': 'IBM', 'start_date': '20240415', 'end_date': 99999999}, 
 {'ticker': 'IBM', 'start_date': '20240417', 'end_date': 99999999}, 
 {'ticker': 'mmm', 'start_date': '20240415', 'end_date': 99999999}, 
 {'ticker': 'mmm', 'start_date': '20240417', 'end_date': 99999999}]

方法2：普通循环写法（更直观） #

如果觉得列表推导式不够清晰，可以用普通循环实现：

new_pos = []
for sublist in positions.values():
    for item in sublist:
        if item['end_date'] == 99999999:
            new_pos.append(item)

效果和嵌套列表推导式完全一致。

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