import os
import pandas as pd
#获取当前目录下的所有txt文件名并存储为列表
folder_path = os.getcwd()
pathlist = [file for file in os.listdir(folder_path) if file.endswith('.txt')]
#提取时间信息
namelist = []
for i in pathlist:
name = i[3:13]
namelist.append(name)
print(pathlist)
print(namelist)
def mkdir(path):
#判断文件夹是否存在
folder = os.path.exists(path)
#如果不存在则创建为文件夹
if not folder:
os.makedirs(path)
print('文件夹创建成功:', path)
else:
print('文件夹已经存在:', path)
#创建中转文件名
#new name list
nnl = []
for i in range(744):
#744表示ueb一个月的输出文件数量
new_name = './new_data/'+str(i)+'.csv'
nnl.append(new_name)
print(nnl)
path = 'C:\\Users\\WLS\\ueb_out\\new_data'
mkdir(path)
for i in range(744):
data = pd.read_table('./'+pathlist[i],encoding='utf-8',header=None,sep=',')
data = data.iloc[:,:3]
df = pd.DataFrame(data)
df['3'] = namelist[i]
df.columns = ['lon','alt','w','time']
df.to_csv(nnl[i])
#每个ueb的输出文件内部共有2051条数据,对应2501个经纬点,计划将744个文件的相同经纬度数据存储在同一个csv文件下
#命名规则:以经纬度命名
#new_csv_name_list
ncnl = []
for i in range(1):
new_df = pd.read_csv()
for j in range(2501):
ncn = str(new_df.iat[j,1])+','+str(new_df.iat[j,2])
#print(ncn)
ncnl.append(ncn)
#print(len(ncnl))
#提取相同不同文件的相同行到新的文件
for i in range(744):
new_df = open(nnl[i],'r')
lines=new_df.readlines()
new_df.close()
lines=lines[1:]
#去表头1:
#print(lines)
for line in lines:
index = lines.index(line)
row = []
row.append(line.split(','))
info = pd.DataFrame(row)
info.to_csv(ncnl[index]+'.csv',mode='a',encoding='utf-8',header=False,index=False)
for i in range(2501):
ff = pd.read_csv(ncnl[i]+'.csv',header=None)
ff.columns = ['sequence','lon','alt','w','time']
max_value_index = ff['w'].idxmax()
max_value_row = ff.loc[max_value_index].values.tolist()
#print(max_value_row)
final_data = pd.DataFrame(max_value_row).transpose()
final_data.to_csv('final_data.csv',mode='a',encoding='utf-8',header=False,index=False)
#跑完之后是一个csv文件,自己再加一个表头,另存为xls版本的excel就可以导入arcgis了