python读取csv文件指定行

最新推荐文章于 2025-05-24 18:36:41 发布

原创最新推荐文章于 2025-05-24 18:36:41 发布 · 7.8k 阅读

7 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#python #numpy #机器学习

收录于

Python 专栏收录该内容

18 篇文章

订阅专栏

该代码示例展示了如何使用Python的csv模块以逐行方式读取CSV文件，并只保留第6行、第7行、第8行、第9行和第10行的每11行数据。通过将这些行存储到numpy数组中，可以有效地处理大型数据集并进行特定行的分析。

import csv
import pandas as pd
import numpy as np

# 这是表示读取前10行
# d = pd.read_csv(file, usecols=['avg_test_acu','avg_test_prec', 'avg_test_TPR', 'avg_test_FPR', 'avg_train_acu'], nrows=10)
k=0
results=[]
with open(file, 'r', newline='', errors='ignore') as f:
    reader = csv.reader(f)
    for index, rows in enumerate(reader):#index从0开始
        if index == 6+k*11 or index == 7+k*11 or index == 8+k*11 or index == 9+k*11 or index == 10+k*11:#目标行
            if index%11==6:#第一折
                a=[]
                a.append(rows)
            elif index % 11 == 10:  # 最后一折
                a.append(rows)
                k=k+1
                results.append(np.array(a))
            else:
                a.append(rows)

results=np.array(results).astype(float)#直接读取的数据是str 转float