Numpy Square All Values In Array Definition