Python Function Change Input Variable And Output