Numpy Create 3d Array From A Lambda