Control screen output precision

control_print_precision.py

import numpy as np
import contextlib

@contextlib.contextmanager
def printoptions(*args, **kwargs):
    original = np.get_printoptions()
    np.set_printoptions(*args, **kwargs)
    yield 
    np.set_printoptions(**original)
    
v = np.random.random(10)
with printoptions(precision=3, suppress=True):
    print(v)