python - 函数，给定一个 csv 文件路径和一个关键字，返回关键字的行号和下一个空行的行号

Question

我正在尝试创建一个函数，给定一个 csv 文件路径和关键字，它可以返回关键字的行号和关键字之后下一个空行的行号。目前，代码按我希望的方式工作，但我遇到了检测空行的问题。该if (row[0] == 0) and (int(reader.line_num) > wordLineNum)部分代码的条件从不测试True它应该的情况（这就是我为blankLineNum上面设置默认值的原因）

def lineFinder (keyword, path): # returns the line number and next blank line within a csv 
                                # file given a keyword
   wordLineNum = 0 #hard coded will remove at some point
   blankLineNum = 8 #will remove once function works as intended
   csvfile = open(path, 'r')
   reader = csv.reader(csvfile)

   for row in reader:
      if str(keyword) in row:
         wordLineNum = int(reader.line_num)

   for row in reader:
      if (row[0] == 0) and (int(reader.line_num) > wordLineNum):
          blankLineNum = int(reader.line_num)

   return wordLineNum , blankLineNum

标签： pythoncsvcsvreader

Answer 1

您的代码将找到文件中的最后一个关键字。这可能是预期的，只是想指出，每次找到关键字时您都会覆盖该值。

在第一个循环的 if 下，您可以添加一个检查：

for row in reader:
    if str(keyword) in row:
        wordLineNum = int(reader.line_num)
    if wordLineNum > 0 and row[0] == 0):
        blankLineNum = int(reader.line_num)
        # Remove the next comment if you want to return the first blank line
        # return wordLineNum , blankLineNum
        
#return the last blank line
return wordLineNum , blankLineNum