[人工智能]python缩减数据

# -*- coding:utf-8 -*-
"""
作者：sunli
日期：2022年04月06日15:36
"""

import numpy as np
import pandas as pd

RAW_DATA_PATH = 'D:/nums.xlsx'


def extract_data(data_file):
    data = pd.read_excel(data_file)
    print(data)
    data_size = len(data.timestep)  # 数量
    print(data_size)
    for i in range(data_size):
        #     # 单元格选取包括df.at[]和df.iat[]两种方法。
        #     # df.at[]和df.iat[]使用时必须输入两个参数，即行索引和列索引，其中df.at[]只能使用标签索引，df.iat[]只能使用整数索引。
        #     # df.at[]和df.iat[]选取的都是单个单元格（单行单列），所以返回值都为基本数据类型
        #
        col_data = np.arange(0, data_size, 10)  # 第一个参数为起点，第二个参数为终点，第三个参数为步长
    print(col_data)
    print(col_data.size)
    extract_data = data.iloc[col_data, [0, 1, 2]]
    print(extract_data)
    extract_data.to_excel('D:/num.xlsx', index=0)


if __name__ == '__main__':
    extract_data(RAW_DATA_PATH)