-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathocean_analysis201.py
260 lines (223 loc) · 8.69 KB
/
ocean_analysis201.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
# -*- coding: utf-8 -*-
import numpy as np
# 实现插值的模块
from scipy import interpolate
#画图的模块
import matplotlib.pyplot as plt
import matplotlib
from pylab import *
#引入相关模块
import requests
import re
import json
import os
import datetime
import time
#setp one 获取港口信息
def get_dir_code(presentation_id):
#定义一个字典用于存放港口名称与港口编号
dir_code_port = {}
# get 访问url,得到json 数据，其实就是字典
url = 'http://oce.ckcest.cn/web/tide/site/querySite.do'
form_data = {
'areaid':presentation_id,
'siteGroupID':3,
}
try:
response = requests.post(url, form_data)
except:
print('没网')
# response 对象的 text 为字符串类型，通过eval()方法转为字典对象。
data_list = eval(response.text)
#拿到data
# print(text)
#获取 showname:code
for data in data_list:
port_id = data['code']
port_name = data['name']
a = {port_name:port_id}
dir_code_port.update(a)
# print(name, name)
print('网络连接较好！')
#数据本地化，以后可以将以下部分改为存储数据库
a=open('港口信息.txt','w+')
a.write(str(dir_code_port))
a.close()
print('港口信息.txt 已写入')
return dir_code_port
def get_ocean_json_data(data):
url = 'http://oce.ckcest.cn/web/knowledge/tide/chaoxi/data/queryDetails.do'
try:
response = requests.post(url,data)
except:
print('没网')
text = response.text
data_json = json.loads(text)
# print(data_json)
file_tile = data_json.get('fileinfo').get('Title')
file_info = data_json.get('fileinfo')
file_data = data_json.get('filedata')
# print(file_data[0].get('Day'))
date = '{}-{}-{}'.format(file_info.get('Year'),file_info.get('Month').zfill(2),str(file_data[0].get('Day')).zfill(2))
file_name= re.sub(r'[() （）]', '', file_tile)+date
# print(file_tile, file_name_f)
# file_name = re.sub(r'[()]', '', '(niggg)')
# print(file_name)
f = open(file_name+'.txt','w+')
for i in range(0,25):
line_time = '{}\t{}:00\t'.format(date, str(i).zfill(2))
line_tide = file_data[0].get('a'+str(i))
line_write = line_time+str(line_tide)+'\n'
f.writelines(line_write)
f.close()
print(file_name+'.txt 文件 已写入')
# 获取y_list return y_list
def get_flie_data(file_name):
f = open(file_name+'.txt')
x= f.readlines()
y_list = []
for i in x:
a,b,c = i.split('\t')
y = eval(c)
if y != None:
y_list.append(y)
# print(y_list)
return y_list
#求出拟合后的水位，yNew
def mtplb(y_list,file_name):
# y 表示纵轴，水位
y = np.array(y_list)
# x是一个横轴，时间（类似时间），从0 到数据端点，间隔为6，表示小时
x_end_point = len(y_list)*6
x = np.array([num for num in range(0,x_end_point,6)])
# 插值法之后的x轴值，表示从0 到 数据端点，每个数据间插入5个值，得出每十分钟
xnew = np.arange(0, x_end_point-6, 1)
"""
kind方法：
nearest、zero、slinear、quadratic、cubic
实现函数func
"""
func = interpolate.interp1d(x, y, kind='cubic')
# 利用xnew和func函数生成ynew，xnew的数量等于ynew数量
ynew = func(xnew)
f = open('{}-plot-tide.txt'.format(file_name),'w+')
for i in ynew:
f.write(format(i,'0.2f')+'\n')
# print(type(ynew))
f.close()
print('文件已写入{}-plot-tide.txt'.format(file_name))
# 画图部分
figsize = 14,6
figure, ax = plt.subplots(figsize = figsize)
zhfont1 = matplotlib.font_manager.FontProperties(fname='SimHei.ttf')
plt.title('{} 潮汐表(24h)'.format(file_name), fontproperties=zhfont1)
plt.xlabel('时刻（h）',fontproperties=zhfont1)
plt.ylabel('潮高（cm）', fontproperties=zhfont1)
plt.tick_params(labelsize=8)
#网格
plt.subplots_adjust(left=0.06,right=1.0, top=0.9, bottom=0.1)
# 原图
plt.plot(x, y, color='g', marker='',label=u"Per 1 hour line")
# 拟合之后的平滑曲线图
# plt.subplot(2,1,1)
plt.plot(xnew, ynew, '.-', color = 'r' ,label=u"Graph per 10 minutes")
# plt.xticks(,[r'$00:00$'])
plt.xticks([num for num in range(0, x_end_point,6)],[r'$00:00$',r'$01:00$',r'$02:00$',r'$03:00$',r'$04:00$',r'$05:00$',r'$06:00$',r'$07:00$',r'$08:00$',r'$09:00$',r'$10:00$',r'$11:00$',r'$12:00$',r'$13:00$',r'$14:00$',r'$15:00$',r'$16:00$',r'$17:00$',r'$18:00$',r'$19:00$',r'$20:00$',r'$21:00$',r'$22:00$',r'$23:00$',],)
plt.legend(loc='upper left')
plt.grid(True)
plt.savefig('{}.png'.format(file_name))
plt.show()
def get_t_file(filename):
f = open('{}-plot-tide.txt'.format(filename))
list_data = f.readlines()
time_str = '{} 00:00'.format(filename[-10:])
# print(time_str)
time_tim = datetime.datetime.strptime(time_str,'%Y-%m-%d %H:%M')
# print(time_tim)
f_t = open('{}.t'.format(filename),'w+')
f_tid = open('{}.tid'.format(filename),'w+')
f_early_eight_tid = open('{}-eight.tid'.format(filename), 'w+')
f_early_eight_tid.write('--------\n')
for data in list_data:
time_w = datetime.datetime.strftime(time_tim,'%H:%M')
tide = format(eval(data)/100,'0.2f')
f_t.write('"{}",{}\n'.format(time_w, tide))
tide_w = '{:0>3d}'.format(int(eval(data)))
f_tid.write('{day}.{moth}.{year} {time}:00 {tied}\n'.format(day = filename[-2:],moth = filename[-5:-3],year = filename[-10:-6],time = time_w,tied =tide_w ))
time_eight=time_tim - datetime.timedelta(hours=8)
f_early_eight_tid.write('{year}/{moth}/{day} {hour}:{minutes}:{sec} {tide}\n'.format(year = time_eight.year, moth=time_eight.month, day=time_eight.day, hour=time_eight.hour, minutes=str(time_eight.minute).zfill(2),sec=str(time_eight.second).zfill(2), tide = tide))
time_tim = time_tim +datetime.timedelta(minutes = 10)
f_t.close()
f_tid.close()
print('{}.t 已写入'.format(filename))
print('{}.tid 已写入'.format(filename))
print('{}-eight.tid 已写入'.format(filename))
#运行函数
def get_port_code_and_data_file():
while True:
f = open('港口信息.txt')
dir_code = eval(f.read())
f.close()
print('请输入以下几个港口名称的一个\n {}, 输入 ‘#’ 返回上一级'.format(str(dir_code.keys())[9:],))
name_port = input('请输入查询港口：')
if name_port == '#':
break
# name_port= '香港'
code = dir_code.get(name_port,'没找到输入的港口')
if code == '没找到输入的港口':
print(code)
else:
while True:
date = input('请输入查询日期格式2019-05-01，输入‘#’号键返回上一级：')
if date == '#':
break
else:
# date = '2019-06-01'
data = {
'PortCode':code,
'Date':date,
'Type':0,
}
get_ocean_json_data(data)
file_name = re.sub(r'[()]','',name_port)+date
y_list = get_flie_data(file_name)
mtplb(y_list, file_name)
get_t_file(file_name)
def exit_print():
print('您已退出程序')
time.sleep(3)
date_end = '2019-12-31'
def awayls_do():
print('\n================潮汐猎人==================\n')
while True:
if time.strptime(date_end,'%Y-%m-%d').__ge__(time.localtime()):
pass
else:
break
dict_presentations = {
"福建":"5359678358229842731",
"广西":"5181903535968969393",
"广州":"5413705892610612211",
"海南":"5625484080778443327",
"河北":"4975833679728738945",
"江苏":"5293027798039637126",
"辽宁":"5432072060831556751",
"山东":"5323306066487582247",
"上海":"5653349478231077601",
"台湾":"4876987809364790815",
"天津":"4935741873097441187",
"浙江":"4845586374601047334"
}
print('请输入以下几个地名中的一个, 输入 ‘exit’ 退出程序：\n\n{}'.format(str(dict_presentations.keys())[9:]))
presentation_name = input('Put in name of presentation:')
# presentation_name='广州'
if presentation_name == 'exit':
exit_print()
break
elif presentation_name not in dict_presentations.keys():
print('地名错误')
else:
pres_id = dict_presentations[presentation_name]
get_dir_code(pres_id)
get_port_code_and_data_file()
awayls_do()