Python Use Output Of Function As Input