numpy转dict

import numpy as np

# 创建一个NumPy数组
arr = np.array([[1, 2], [3, 4]])

# 将NumPy数组转换为Python列表
arr_list = arr.tolist()

# 使用列表推导式将列表转换为字典
arr_dict = {index: value for index, value in enumerate(arr_list)}

print(arr_dict)

{0: [1, 2], 1: [3, 4]}

import numpy as np

# 创建一个多维NumPy数组
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

# 将NumPy数组转换为Python列表
arr_list = arr.tolist()

# 使用多层列表推导式将列表转换为嵌套字典
arr_dict = {index1: {index2: value for index2, value in enumerate(inner_list)} for index1, inner_list in enumerate(arr_list)}

print(arr_dict)

{0: {0: [1, 2], 1: [3, 4]}, 1: {0: [5, 6], 1: [7, 8]}}