Python Copy A Function In Math