如何更Pythonic地获取字典任意元素以初始化对应维度numpy数组

核心最优写法 #

直接通过迭代器获取字典的第一个值，无需转换整个键列表，既Pythonic又高效：

import numpy as np

# 假设embedding是你的字典，值为numpy数组形式的embedding
arr = np.zeros(next(iter(embedding.values())).shape)

为什么这比原来的写法好 #

• 效率更高：iter(embedding.values())创建一个值的迭代器，next()只取第一个元素，不会像list(embedding)那样把所有键加载到内存里，尤其当字典规模很大时，内存占用差异明显。
• 更简洁直观：直接操作值的迭代器，符合Python"迭代优先"的设计哲学，代码逻辑更清晰。

处理空字典的边界场景 #

如果需要兼容字典为空的情况，可以添加简单判断：

if not embedding:
    raise ValueError("Embedding字典不能为空")
arr = np.zeros(next(iter(embedding.values())).shape)

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