Python3 picture affine transformation