Python中如何在循环中单独访问字典内每个array的指定位置元素？

嘿，我来帮你解决这个问题！你的字典里每个值都是NumPy数组对吧？要单独访问每个数组的第1、2、3...个元素其实没那么复杂，我给你几个实用的实现方式：

方法1：直接遍历字典值，按索引取元素 #

Python里字典的values()方法可以直接拿到所有数组，然后你只要针对每个数组用索引（注意是从0开始的哦，第一个元素是索引0，第二个是1，以此类推）就能取出对应元素：

import numpy as np

# 你的目标字典
my_dict = {0: np.array([0, 0, 0, 0, 0]), 1: np.array([0, 1, 0, 0, 0]), 2: np.array([0, 0, 1, 0, 0]), 3: np.array([0, 0, 0, 0, 0]), 4: np.array([0, 0, 0, 1, 0]), 5: np.array([0, 0, 0, 0, 0]), 6: np.array([0, 0, 0, 0, 0]), 7: np.array([0, 0, 0, 0, 0]), 8: np.array([0, 0, 0, 0, 0]), 9: np.array([0, 0, 0, 0, 0])}

# 示例：取每个数组的第一个元素（索引0）
for arr in my_dict.values():
    first_element = arr[0]
    print(first_element)  # 换成你需要的操作就行

# 如果要取第二个元素，把索引改成1就行
for arr in my_dict.values():
    second_element = arr[1]
    print(second_element)

方法2：同时获取键和对应元素 #

如果需要知道每个元素来自字典的哪个键，可以用items()方法同时遍历键和数组：

# 示例：取每个数组的第三个元素（索引2）
for key, arr in my_dict.items():
    third_element = arr[2]
    print(f"键{key}对应的数组第三个元素是：{third_element}")

方法3：批量收集所有数组的第n个元素 #

要是你想一次性把所有数组的某一位元素整理成列表，用列表推导式会更高效：

# 示例：收集所有数组的第四个元素（索引3）
fourth_elements = [arr[3] for arr in my_dict.values()]
print(fourth_elements)  # 输出：[0, 0, 0, 0, 1, 0, 0, 0, 0, 0]

顺便说一句，你之前尝试把字典值转成列表的思路是对的，比如arr_list = list(my_dict.values())，之后遍历这个列表同样能按索引取元素，可能是之前的循环写法没到位？比如这样也是可行的：

arr_list = list(my_dict.values())
for arr in arr_list:
    print(arr[0])  # 同样能拿到每个数组的第一个元素

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