省略K长度行的Python程序