如何遍历存储在变量 python 中的文件中的行

Question

我有一个 html 形式的解码文件存储在一个变量中：

file_data = file.read().decode('utf-8')
print(file_data)

我想用一个跟踪行号的计数器遍历文件，以便我可以获得表开始时的行号。 这是我尝试过的两种方法：

方法一：

counter = 0
for row in file_data:
    if row == 'Date,Unique Id,Tran Type,Cheque Number,Payee,Memo,Amount':
        date_line_no = counter
        break
    counter += 1

方法二：

for line, row in enumerate(file_data):
    first_column = row[0]
    if first_column == 'Date':
        print(row)
        date_line_no = line

我理想的 output 将是 7，因为那是表格从其列开始的时候。

Answer 1

使用enumerate function，然后搜索包含逗号的行，如果它确实用逗号分隔，然后继续您当前的方式。 从文件中读取时，您还应该使用with语句。 这避免了必须立即读取整个文件并确保它在最后正确关闭文件。

例如：

with open(filename) as file:
    for i,row in enumerate(file):
        if ',' in row and row.split(',')[0] == 'Date':
            break
print(i)  # output is the index of the header row.

如果您必须以自己的方式打开文件，那么它仍然是相同的，除了您还需要将整个文件拆分为\n 。

file_data = file.read().decode('utf-8')
print(file_data)


for i, row in enumerate(file_data.split('\n')):
    if ',' in row and row.split(',')[0] == 'Date':
        break
print(i)

Answer 2

# Opening the file
file = open('your_file_name_or_location') 

# adding a counter for each line
for index, content in enumerate(file):

    var = content.split(',')   # Spliting each line by ','
    if var[0] == 'date':       
        break                  # Getting out of loop if first element is 'date'
print(index + 1)  # Position of header of the table