pandas 批量修改列名_基于Pandas的给批量文件增加列名的算法

import os

import pandas as pd

import numpy as np

import math

# 获取文件名

def get_data_set(path):

data_directory = path + "/data"

data_files = []

files = os.listdir(data_directory)

files.sort()

for f in files:

try:

data_files.append(f)

except Exception:

msg = traceback.format_exc()

print (msg)

print('Warning: unknown file', f)

return data_files #返回一个文件名列表

#根据文件名读文件

def data_read(path,file_name):

data_file_path = path + "/data/"

data_file_path = data_file_path + file_name

name,extension = file_name.split(".")

if extension == "csv":

try:

data = pd.read_csv(data_file_path,encoding = "gbk",header = None)

except:

dat