Python Input Variable In Functions