python 读入csv跳过表头

如题。
在读入csv文件时,有时我们指向读入数据并把它赋给矩阵,那么如何跳过表头?
方法如下:

import csv
from itertools import islice
...
...

with open('datasets/diabetes.csv') as data_file:
    data_csv = csv.reader(data_file)
    row_num = 0
    for row in islice(data_csv, 1, None):
        for i in range(len(row)):
            row[i] = float(row[i])
        label_set[row_num] = row[-1]
        row[-1] = 1
        # print(row)
        data_set[row_num] = row
        row_num = row_num + 1

使用islice即可。

更多推荐

python 读入csv跳过表头