你可以使用 NumPy 的 around 函数来保留 ndarray 数据的小数位数。例如,要将 ndarray 中的所有元素保留两位小数,你可以这样写:
import numpy as np
# 假设 x 是一个 ndarray
x = np.array([1.234567, 2.345678, 3.456789])
# 保留两位小数
x = np.around(x, decimals=2)
print(x) # 输出 [1.23 2.35 3.46]
注意,around 函数会使用四舍五入的方式来保留小数位数。如果你想使用其他的方式来保留小数位数,你可以使用 N