python将数组n等分的实例