Create Graph From Function Python