Shape Function In Python