你可以set_printoptions
用来设置输出的精度:
import numpy as np
x=np.random.random(10)
print(x)
# [ 0.07837821 0.48002108 0.41274116 0.82993414 0.77610352 0.1023732
# 0.51303098 0.4617183 0.33487207 0.71162095]
np.set_printoptions(precision=3)
print(x)
# [ 0.078 0.48 0.413 0.83 0.776 0.102 0.513 0.462 0.335 0.712]
并suppress
禁止对小数使用科学计数法:
y=np.array([1.5e-10,1.5,1500])
print(y)
# [ 1.500e-10 1.500e+00 1.500e+03]
np.set_printoptions(suppress=True)
print(y)
# [ 0. 1.5 1500. ]
有关其他选项,请参见文档中的set_printoptions
。
要使用NumPy 1.15.0或更高版本在本地应用打印选项,可以使用numpy.printoptions
上下文管理器。例如,在with-suite precision=3
和suppress=True
内设置:
x = np.random.random(10)
with np.printoptions(precision=3, suppress=True):
print(x)
# [ 0.073 0.461 0.689 0.754 0.624 0.901 0.049 0.582 0.557 0.348]
但是在with-suite打印选项之外,将恢复为默认设置:
print(x)
# [ 0.07334334 0.46132615 0.68935231 0.75379645 0.62424021 0.90115836
# 0.04879837 0.58207504 0.55694118 0.34768638]
如果你使用的是NumPy的早期版本,则可以自己创建上下文管理器。例如,
import numpy as np
import contextlib
@contextlib.contextmanager
def printoptions(*args, **kwargs):
original = np.get_printoptions()
np.set_printoptions(*args, **kwargs)
try:
yield
finally:
np.set_printoptions(**original)
x = np.random.random(10)
with printoptions(precision=3, suppress=True):
print(x)
# [ 0.073 0.461 0.689 0.754 0.624 0.901 0.049 0.582 0.557 0.348]
为防止浮点数结尾处的零被剥离:
np.set_printoptions
现在有一个formatter
参数,可让你为每种类型指定格式功能。
np.set_printoptions(formatter={'float': '{: 0.3f}'.format})
print(x)
哪个打印
[ 0.078 0.480 0.413 0.830 0.776 0.102 0.513 0.462 0.335 0.712]
代替
[ 0.078 0.48 0.413 0.83 0.776 0.102 0.513 0.462 0.335 0.712]