Python中如何将Numpy数组转换为键值均四舍五入取整的字典

核心思路 #

对数组每行的两个元素分别做四舍五入取整，第一个元素转字符串作为字典键，第二个元素作为字典值。

完整代码 #

import numpy as np

# 定义原始Numpy数组
origin_arr = np.array([
    [8.82847075, -5.70925653],
    [1.07032615, -1.77975378],
    [-10.41163742, -0.33042086],
    [0.23799394, 5.5978591],
    [7.7386861, -4.16523845]
])

# 转换为目标字典
result_dict = {str(int(np.round(row[0]))): int(np.round(row[1])) for row in origin_arr}

# 验证输出
print(result_dict)

输出结果 #

{'9': -6, '1': -2, '-10': 0, '0': 6, '8': -4}

说明 #

• 该写法完全兼容Python 3.10版本，仅需提前安装Numpy库即可运行
• np.round() 按照标准四舍五入规则取最近整数，转换为int是为了去除numpy浮点类型后缀
• 需确保输入数组结构为n行2列，每行第一个元素对应键、第二个对应值

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