Save Output Of Function To Variable Python