import pandas as pd from tqdm import tqdm # 读取NGSIM数据文件，假设数据文件名为ngsim_data.csv data = pd.read_csv(r'F:\PycharmProjects\ngsim\dataset\v_class===2.xlsx') # 创建一个进度条，指定总的迭代次数 progress_bar = tqdm(total=len(data)) # 存储结果的列表 result = [] # 处理数据 for index, row in data.iterrows(): # 判断第十四列的值是否在1、2、3、4、5之间 if row[14] in [1, 2, 3, 4, 5]: # 在这里进行数据处理，例如打印行数据或执行其他操作 # 假设将行数据存储到result列表中 result.append(row.tolist()) # 更新进度条 progress_bar.update(1) # 完成后关闭进度条 progress_bar.close() # 将结果存储到文件 with open('Line_ID_1-5.txt', 'w') as file: for row_data in result: file.write(','.join(map(str, row_data)) + '\n')

你的代码基本上是正确的，它读取了名为 `ngsim_data.csv` 的数据文件，并在第十四列的值为 1、2、3、4 或 5 的行进行处理。处理后的结果存储在 `result` 列表中，并将结果写入名为 `Line_ID_1-5.txt` 的文件中。代码中还添加了一个进度条，以显示处理进度。

不过，需要注意的是，你的数据文件似乎是一个 Excel 文件，而不是一个 CSV 文件。因此，在 `pd.read_csv()` 中使用 `.xlsx` 扩展名可能会导致错误。你可以尝试使用 `pd.read_excel()` 函数来读取 Excel 文件。另外，还需要确保已经安装了相应的依赖库 `pandas` 和 `tqdm`。

以下是稍作修改后的代码示例：

import pandas as pd
from tqdm import tqdm

# 读取NGSIM数据文件，假设数据文件名为ngsim_data.xlsx
data = pd.read_excel(r'F:\PycharmProjects\ngsim\dataset\v_class===2.xlsx')

# 创建一个进度条，指定总的迭代次数
progress_bar = tqdm(total=len(data))

# 存储结果的列表
result = []
# 处理数据
for index, row in data.iterrows():
    # 判断第十四列的值是否在1、2、3、4、5之间
    if row[14] in [1, 2, 3, 4, 5]:
        # 在这里进行数据处理，例如打印行数据或执行其他操作
        # 假设将行数据存储到result列表中
        result.append(row.tolist())

    # 更新进度条
    progress_bar.update(1)

# 完成后关闭进度条
progress_bar.close()

# 将结果存储到文件
with open('Line_ID_1-5.txt', 'w') as file:
    for row_data in result:
        file.write(','.join(map(str, row_data)) + '\n')

请确保将文件路径 `F:\PycharmProjects\ngsim\dataset\v_class===2.xlsx` 替换为你实际的数据文件路径，并确保已安装所需的依赖库。