Skip to content

Commit

Permalink
demo code
Browse files Browse the repository at this point in the history
  • Loading branch information
emfdavid committed Feb 27, 2024
1 parent 1ea1003 commit 1c6d5b6
Showing 1 changed file with 51 additions and 0 deletions.
51 changes: 51 additions & 0 deletions demo.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
import logging

logging.basicConfig(
format="%(asctime)s.%(msecs)03dZ %(processName)s %(threadName)s %(levelname)s:%(name)s:%(message)s",
datefmt="%Y-%m-%dT%H:%M:%S",
level=logging.INFO,
)

logger = logging.getLogger("grib demo")

import xarray as xr
import numpy as np
import pandas as pd
import zarr
import kerchunk
import fsspec
import gcsfs


if __name__ == "__main__":

fs = fsspec.filesystem(
protocol="reference",
fo="gs://gcp-public-data-weather.camus.store/high-resolution-rapid-refresh/version_2/monthly_horizon/conus/hrrr.202309/hrrr.wrfsfcf.16_hour_horizon.zarr", # The target zarr json blob
remote_protocol="gcs",
)


ds = xr.open_dataset(
fs.get_mapper(""),
engine="zarr",
drop_variables=["heightAboveGround"], # Drop magic broken variable
backend_kwargs=dict(
consolidated=False,
)
)
logger.info(ds)

df = ds[
['2t', 'dswrf']
].interp(
dict(x=[500, 800], y=[300, 400])
).to_dataframe()

# .loc[
# dict(valid_time=slice('2023-09-04', '2023-09-06'))
# ].compute()



logger.info(df)

0 comments on commit 1c6d5b6

Please sign in to comment.