Add None Dimension To Numpy Function Python Code