python算法4