python - 如何舍入numpy数组中的特定数字选择？

Question

从随机 numpy 列表中，我只想舍入 padInputs 索引中的数字。以下代码是我正在尝试但不起作用的代码。什么是解决方法？

padInputs = [0, 2, 7, 8]
random = np.random.rand(13)
for padInput in padInputs:
    np.around(random[padInput])

例如，

Input
[0.87720789, 0.88194004, 0.06039337, 0.13874861, 0.85552875]

Output
[0.87720789, 1, 0, 0.13874861, 0.85552875]

Answer 1

试试这个方法:

random[padInputs] = np.around(random[padInputs])

请注意，这将不带小数点四舍五入，您可以通过以下方式将其作为参数传递给四舍五入:

random[padInputs] = np.around(random[padInputs], decimals=2)