aboutsummaryrefslogtreecommitdiff
path: root/works/life/gpcp/GpcpConverter.py
diff options
context:
space:
mode:
authorYuqian Yang <crupest@crupest.life>2025-02-12 15:55:21 +0800
committerYuqian Yang <crupest@crupest.life>2025-02-12 15:55:21 +0800
commit1ecfd0ab7f1f511268fd6404dbc110c3c277b48c (patch)
tree49449a4076ded9bd937a51679318edbe2a532cae /works/life/gpcp/GpcpConverter.py
parent55d8b025e8d6ea971e8ee5762c892405fedc316b (diff)
parentf8c10dd1fc55e60f35286475356e48c4f642eb63 (diff)
downloadcrupest-1ecfd0ab7f1f511268fd6404dbc110c3c277b48c.tar.gz
crupest-1ecfd0ab7f1f511268fd6404dbc110c3c277b48c.tar.bz2
crupest-1ecfd0ab7f1f511268fd6404dbc110c3c277b48c.zip
import(life): IMPORT crupest/life COMPLETE.
Diffstat (limited to 'works/life/gpcp/GpcpConverter.py')
-rw-r--r--works/life/gpcp/GpcpConverter.py29
1 files changed, 29 insertions, 0 deletions
diff --git a/works/life/gpcp/GpcpConverter.py b/works/life/gpcp/GpcpConverter.py
new file mode 100644
index 0000000..7aadd44
--- /dev/null
+++ b/works/life/gpcp/GpcpConverter.py
@@ -0,0 +1,29 @@
+import pandas
+from pandas import DataFrame
+import xarray as xr
+import os
+import os.path
+
+latitude = 30
+longitude = 114
+
+data_dir = os.path.join(os.path.dirname(__file__), "GpcpData")
+files = os.listdir(data_dir)
+files = [os.path.join(data_dir, f) for f in files if f.endswith(".nc")]
+files.sort()
+
+result = DataFrame([], columns=["date", "prec"], dtype="object")
+
+for file in files:
+ data_set = xr.open_dataset(file)
+ df = data_set.to_dataframe()
+ data_set.close()
+ df = df.query(
+ f"latitude == {latitude} & longitude == {longitude} & nv == 1")
+ df = df.reset_index()
+ df = df.drop(columns=["latitude", "longitude", "nv",
+ "lat_bounds", "lon_bounds", "time_bounds"])
+ df = df.rename(columns={"time": "date", "precip": "prec"})
+ result = pandas.concat([result, df], ignore_index=True)
+
+result.to_csv("./out.csv")