python得到数组前几行