mengke25’s python function