File size: 1,051 Bytes
d2f661a |
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 |
import os
import pyart
import pathlib
from fire import Fire
def label_all_items(
data_dir="/data/data_WF/NhaBe/2022",
out_dir="/data/data_WF/NhaBe/2022"
):
for subdir in os.listdir(data_dir):
subdir_path = os.path.join(data_dir, subdir)
for i in range (1,32):
if i < 10:
data_new_dir = subdir_path + '/0{}'.format(i)
out_new_dir = subdir_path + '/0{}'.format(i)
else:
data_new_dir = subdir_path + '/{}'.format(i)
out_new_dir = subdir_path + '/{}'.format(i)
pathlist = pathlib.Path(data_new_dir).rglob('*.*')
for path in pathlist:
f = os.path.join(path)
radar = pyart.io.read(f)
t = radar.time['units'].split(" ")[2][:-4]
outdir = pathlib.Path(out_new_dir)
t = outdir/t
os.rename(f, t)
# print(t)
# print(data_new_dir)
if __name__ == "__main__":
Fire(label_all_items)
|