如何用更Pythonic的方式从可变列文件生成字典？

一、生成目标格式字典的更Pythonic写法 #

要生成{'a': [1, 2, 3], 'b': 2, 'c': 3}这种“多元素存列表、单元素存值”的字典，核心是避免重复分割字符串，同时兼顾可读性和效率，推荐两种写法：

写法1：高效易读的循环（优先推荐） #

循环写法看似不简洁，但可读性拉满，且仅做一次字符串分割，性能更优：

with open('data_vec.txt') as f:
    target_dict = {}
    for line in f:
        parts = line.strip().split()
        if not parts:  # 跳过空行
            continue
        key, *rest = parts
        # 多元素存列表，单元素存值，无元素则设为None（可按需调整）
        target_dict[key] = rest if len(rest) > 1 else rest[0] if rest else None

写法2：简洁的字典推导式 #

如果追求一行式（可读性稍降），可以用字典推导式配合海象运算符避免重复分割：

with open('data_vec.txt') as f:
    target_dict = {
        key: rest if len(rest) > 1 else rest[0] if rest else None
        for line in f
        if (parts := line.strip().split())
        for key, *rest in [parts]
    }

二、验证foo = dict([[line.split()[0], line.split()[1:]] for line in file])的可行性 #

这个写法语法上完全可行，但存在两个明显问题：

1. 效率浪费：每一行调用两次line.split()，重复处理同一个字符串，大文件场景下性能损耗明显；
2. 不符合目标格式：无论后面是单个还是多个元素，所有值都会被转为列表。比如你期望'b': 2，实际会得到'b': [2]，和需求结构不符。

如果你的需求允许所有值为列表类型，这个写法能正常生成字典，但显然不匹配你给出的目标格式。

内容的提问来源于stack exchange，提问作者G. Gare