python循环数组判断