# !/usr/bin/env python # -*- coding:utf-8 -*- # author: haotian time:2019/9/14 import numpy as np f = open("./data/CD_Flight190914A.csv", "rb") excel = open("./data/time_flight.csv", "w+") # position_exl = open("./data/position_exl.csv", "w+") schedule = np.loadtxt(f, dtype=str, delimiter=",", skiprows=1, usecols=(4,)) # 分隔符 空格 Array = np.zeros(209) count = 1 i = 0 n = 0 while i < (len(schedule)-1): if schedule[i] == schedule[i + 1] : # 如果航班时间重复 创建一个不重复的时间表记录重复次数 count = count + 1 else: Array[n] = count #Array存的重复次数 count = 0 n = n + 1 i = i + 1 new_schedule,a = np.unique(schedule,return_index=True) #去掉相同时间的数据 # for i in range(len(position)): # position_exl.write(str(position[i])+',\n') # position_exl.close() # position_exl = open(("./data/position_exl.csv", "w+")) # position = np.loadtxt(position_exl, dtype=float, delimiter=",", skiprows=0, usecols=(0,)) # new_schedule = [len(position)*''] # n = 0 # numbers = [ int(x) for x in position ] # for i in range(numbers): # new_schedule[n] = schedule[i] # n = n + 1 excel.write("Schedule,PlaneNum"+'\n') for i in range(len(new_schedule)-1): excel.write(str(new_schedule[i])+","+str(Array[i])+",\n") excel.close() ''' 此时的数据time_flight.csv由于排序的原因导致时间的序列不一致, 最终数据用excel降序排列并保存到schedule_PlaneNum.csv中 '''完善并使其运行
时间: 2024-01-07 19:03:10 浏览: 72
这段代码是用来处理航班时间表的,读取 CD_Flight190914A.csv 文件中的航班时间数据,统计每个时间段内的航班数目,然后将数据保存到 time_flight.csv 文件中。其中,new_schedule 保存去重后的时间表,Array 数组保存每个时间段内的航班数目。
代码存在一些问题,比如:
1. 在写入文件时,每行最后多了一个逗号,需要去掉。
2. position_exl 文件的路径错误,应该是 "./data/position_exl.csv"。
3. position_exl 文件没有正确写入,应该是使用 f.write() 方法写入数据。
4. 关于 position_exl 文件的处理,代码中存在一些注释,但是没有被执行,需要根据需要进行修改。
修改后的代码如下:
```python
#!/usr/bin/env python
# -*- coding:utf-8 -*-
# author: haotian time:2019/9/14
import numpy as np
f = open("./data/CD_Flight190914A.csv", "rb")
excel = open("./data/time_flight.csv", "w+")
position_exl = open("./data/position_exl.csv", "w+")
schedule = np.loadtxt(f, dtype=str, delimiter=",", skiprows=1, usecols=(4,)) # 分隔符 空格
Array = np.zeros(209)
count = 1
i = 0
n = 0
while i < (len(schedule)-1):
if schedule[i] == schedule[i + 1]:
count = count + 1
else:
Array[n] = count
count = 0
n = n + 1
i = i + 1
# 将时间表去重
new_schedule, a = np.unique(schedule, return_index=True)
# 将位置信息写入到 position_exl 文件中
for i in range(len(new_schedule)):
position_exl.write(str(i+1) + ',\n')
position_exl.close()
# 重新打开 position_exl 文件,并读取数据
position_exl = open("./data/position_exl.csv", "r")
position = np.loadtxt(position_exl, dtype=float, delimiter=",", skiprows=0, usecols=(0,))
# 将航班时间和航班数目写入到 excel 文件中
excel.write("Schedule,PlaneNum\n")
for i in range(len(new_schedule)):
excel.write(str(new_schedule[i])+","+str(Array[i])+"\n")
excel.close()
# 将 excel 文件降序排列
data = np.loadtxt("./data/time_flight.csv", dtype=str, delimiter=",", skiprows=1)
data = data[data[:, 1].astype(int).argsort()[::-1]]
# 将排好序的数据保存到 schedule_PlaneNum.csv 文件中
np.savetxt("./data/schedule_PlaneNum.csv", data, delimiter=",", fmt="%s")
```
阅读全文