python提取csv指定列

# -*- coding:utf-8 -*-
"""
作者：sunli
日期：2022年03月09日21:19
"""

import os
import cv2
import pandas as pd
import numpy as np
import configparser as cp
import matplotlib.pyplot as plt


PATH='D:/FLDetection/Data/t'


def extract_data(data_file):
    """
    从mobileFall中提取数据，用于做实验测试
    :param data_file:  原始数据文件
    :param sampling_frequency: 原始数据采集频率
    :return:
    """
    data = pd.read_csv(data_file)  # 指定源文件第一列作为行索引
    data_size = len(data.label)  # 标签数量
    for i in range(data_size):
        col_data = np.arange(0, data_size, 1)  # 第一个参数为起点，第二个参数为终点，第三个参数为步长
        # # 滑动窗口就是能够根据指定的单位长度来框住时间序列，从而计算框内的统计指标。相当于一个长度指定的滑块在刻度尺上面滑动，每滑动一个单位即可反馈滑块内的数据。
    extract_data = data.iloc[col_data, [2, 3, 4, 5, 6, 7, 8]]  # 选取加速度计和陀螺仪数据和行为标签
    print(extract_data)

    # save_path = os.path.abspath('D:/shujuji/MobiAct_Dataset_v2.0/Annotated Data/DATA/'