如何选择前3行，然后添加第四行(如果有)，直到n-1

def select_rows(n):
    rows = []  # 存储选中的行
    count = 0  # 已选中的行数

    for i in range(1, n):
        if count < 3:
            rows.append(i)
            count += 1
        elif count == 3:
            rows.append(i)
            count += 1
            break
        else:
            break
    
    if n > 3:
        for i in range(4, n):
            rows.append(i)
            count += 1
    
    return rows

n = 10  # 给定的行数
selected_rows = select_rows(n)
print(selected_rows)