import numpy as n