import json
import csv
import pandas as pd
# 循环读取当前目录下所有csv文件
import os
n = 117768
list = os.listdir('./data_10km_tp/') #列出文件夹下所有的目录与文件
for i in range(1,len(list)):
path = os.path.join('./data_10km_tp/',list[i])
path2 = os.path.join('./data_10km_tp/',list[i-1])
if os.path.isfile(path):
print(path)
# 读取csv文件
data = pd.read_csv(path,encoding='utf-8')
# 将csv文件转换为json格式
else:
print('不是文件!')
if os.path.isfile(path2):
print(path2)
# 读取csv文件
data2 = pd.read_csv(path2,encoding='utf-8')
else:
print('不是文件!')
save_data = data2.copy()
# 遍历行json数据
for j in range(0, len(data2)):
if data2.iloc[j]["id"] in data["id"].values:
save_data.loc[j, "XCoordinate"] = (data[data["id"] == data.iloc[j]["id"] ]["XCoordinate"].values + data2.loc[j, "XCoordinate"])/2
save_data.loc[j,"YCoordinate"] = (data[data["id"] == data.iloc[j]["id"] ]["YCoordinate"].values + data2.loc[j, "YCoordinate"])/2
# json保存为新的csv文件,并添加表头
df = pd.DataFrame(data)
df.to_csv('./new_data/{}'.format(n)+".csv",encoding='utf-8',index=False)
n = n+1
df = pd.DataFrame(save_data)
df.to_csv('./new_data/{}'.format(n)+".csv",encoding='utf-8',index=False)
n = n+1
df = pd.DataFrame(data2)
df.to_csv('./new_data/{}'.format(n)+".csv",encoding='utf-8',index=False)
n = n+1
【Python】Pandas行跟列的处理
最新推荐文章于 2024-04-17 09:41:39 发布