在python中第一次出現空行csv後查找並刪除

Question

我想在第一次出現空行後刪除所有行。

1. 什麼是最好的方式來讀取行直到遇到一個空行？
2. 什麼是刪除所有正在進行的行的最佳命令？

Answer 1

這是您可以使用的腳本。主函數中的代碼只是爲了讓您選擇一個文件（或者如果給出，可以將其作爲命令行參數選取）。然後調用crop_file函數，通過打開文件，將數據讀到第一個空行到內存中，關閉文件，然後將存儲的數據寫入同一個文件，如果確認需要剩餘的行被刪除。

讓我知道如果您有任何關於代碼的問題。

#!/usr/bin/python 

import os, csv, sys, Tkinter, tkFileDialog as fd 

# stop tinker shell from opening as only needed for file dialog 
root = Tkinter.Tk() 
root.withdraw() 

def crop_file(in_path): 
    # read file into memory 
    file_obj = open(in_path, 'rb') 
    reader = csv.reader(file_obj, delimiter='\t') 
    data = [] 
    for row in reader: 
     if not row or not any(row): 
      break #stop at empty row 
     else: 
      data.append(row) 
    file_obj.close() 

    print 'Found', len(data), 'rows of data without empty lines.' 
    conf = raw_input('delete remaining lines? (Y|N): ').upper()[0] 

    if conf == 'Y': 
     # write data to file 
     file_obj = open(in_path, 'wb') 
     writer = csv.writer(file_obj) 
     writer.writerows(data) 
     file_obj.close 

def main(): 
    in_path = None 
    prog_name = sys.argv[0] 

    # check if in_path are inlcuded as cmd line args... 
    if len(sys.argv) > 1: 
     in_path = sys.argv[1] 
     if not os.path.exists(in_path): 
      print 'Usage:', prog_name, '[file_path>]' 
      print 'cannot find the file provided for file_path:\n', in_path 
      sys.exit("Error - invalid excel_file_path arg") 
    else: 
     try: 
      # set current working directory to user's my documents folder 
      os.chdir(os.path.join(os.getenv('userprofile'),'documents')) 
     except: 
      pass 

    # ask user for path to file... 
    while not in_path: 
     print "Please select the file to read data from ..." 
     try: 
      in_path = fd.askopenfilename() 
     except: 
      print 'Error selecting file.' 
     if not in_path: 
      cont = raw_input('Do you want to continue? (Y|N): ').upper()[0] 
      if cont == 'N': 
       sys.exit("Error - unable to select input file") 

    crop_file(in_path) 

if __name__ == '__main__': 
    main()