Reduce Python Geeksforgeeks Numpy