如何找到包含给定字符串的行索引？ Python

Question

我想升级我的脚本来分析数据。 我需要找到包含特定字符串的索引行，而不是手动检查行号作为标题行。 现在我直接将 csv 读取到 pandas 数据帧，标题行定义如下：

df1 = pd.read_csv('sensor_1.csv', sep=',', header=101)

如何读取 csv 并找到带有“扫描编号”文本的行并将此变量放入标题定义中？

我试过这个：

FileList = (glob.glob("sensor_1.csv"))
for FileToProcess in FileList:
        with open(FileToProcess) as readfile:
            for cnt,line in enumerate(readfile):
                if "Scan Number" in line:
                    cnt
        readfile.close

df1 = pd.read_csv('sensor_1.csv', sep=',', header= cnt)

但这最后给出了最高的索引和错误：/你能帮忙吗？

谢谢宝琳娜

Answer 1

fille_ =  open('sensor_1.csv', 'r')
lines = fille_.readlines()
cnt = 0
for i in range(0, len(lines)):
    if lines[i].find('Scan Number') !=-1:
        cnt = i
        break
        
print(cnt)

当在字符串中找到搜索短语时，循环将打印字符串的索引并停止循环。

Answer 2

你不需要使用它用于在目录中列出文件的地球库......但你已经有了文件名

当你使用'with'命令时，你也不需要关闭文件。 缩进后会关闭

尝试这个

import pandas as pd , csv

rownumber_list = []
with open("Salary_Data.csv", 'r') as file:
    csvreader = csv.reader(file)
    for cnt,line in enumerate(csvreader):
        if "Scan Number" in line:
            rownumber_list.append(cnt)

df1 = pd.DataFrame(rownumber_list , columns=['row_number'])