python for range array length