skyborn 0.1.0__tar.gz

skyborn-0.1.0/PKG-INFO

@@ -0,0 +1,28 @@
+Metadata-Version: 2.2
+Name: skyborn
+Version: 0.1.0
+Summary: Atmospheric science research utilities
+Author: Qianye Su
+Author-email: suqianye2000@gmail.com
+License: MIT
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Dist: numpy>=1.20.0
+Requires-Dist: xarray>=0.19.0
+Requires-Dist: matplotlib>=3.4.0
+Requires-Dist: cartopy>=0.20.0
+Requires-Dist: netCDF4>=1.5.7
+Requires-Dist: metpy>=1.1.0
+Requires-Dist: cfgrib>=0.9.9
+Requires-Dist: eccodes>=1.4.0
+Requires-Dist: scikit-learn>=1.0.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: summary
+
+Skyborn is a tool for easy plotting ERA5 weather data.

skyborn-0.1.0/README.md

@@ -0,0 +1,11 @@
+# Skyborn: A Collection of Tools for My Research
+
+Skyborn is a collection of functions I use in my general research, primarily for personal use.
+
+## Installation
+
+To install the Skyborn package, you can use pip:
+
+```bash
+pip install skyborn
+```

skyborn-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

skyborn-0.1.0/setup.py

@@ -0,0 +1,30 @@
+import setuptools
+
+setuptools.setup(
+    name="skyborn",
+    version="0.1.0",  # 与pyproject.toml保持一致
+    author="Qianye Su",
+    author_email="suqianye2000@gmail.com",
+    description="Atmospheric science research utilities",
+    long_description="Skyborn is a tool for easy plotting ERA5 weather data.",
+    license="MIT",
+    packages=setuptools.find_packages(where="src"),
+    package_dir={"": "src"},
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+    ],
+    license_files=("LICENSE.txt"),
+    install_requires=[
+        "numpy>=1.20.0",
+        "xarray>=0.19.0",
+        "matplotlib>=3.4.0",
+        "cartopy>=0.20.0",
+        "netCDF4>=1.5.7",
+        "metpy>=1.1.0",
+        "cfgrib>=0.9.9",
+        "eccodes>=1.4.0",
+        "scikit-learn>=1.0.0"
+    ]
+)

skyborn-0.1.0/src/skyborn/__init__.py

@@ -0,0 +1,21 @@
+"""
+Skyborn 大气科学研究工具库
+
+提供大气科学常用数据处理、分析和可视化功能模块
+"""
+
+from .io import read_netcdf, read_grib
+from .calculations import (
+    convert_longitude_range,
+    linearRegression
+)
+from .plotting import add_equal_axes, createFigure
+from .gradients import (
+    calculate_gradient,
+    calculate_meridional_gradient,
+    calculate_zonal_gradient,
+    calculate_vertical_gradient
+)
+
+__version__ = "0.1.0"
+__all__ = ['io', 'calculations', 'gradients', 'plotting']

skyborn-0.1.0/src/skyborn/calculations.py

@@ -0,0 +1,54 @@
+import numpy as np
+import xarray as xr
+import metpy.calc as mpcalc
+from metpy.units import units
+from typing import Tuple, Union
+from sklearn.feature_selection import f_regression
+
+
+def linearRegression(mat, sequence):
+    '''
+    回归函数 第一个是需要回归的变量，第二个为一个数组
+    '''
+    if len(mat) != len(sequence):
+        raise ValueError('Data, array must be must be equal!!!!!!!'
+                         ' %s and %s' % (mat.shape[0], len(sequence)))
+    A = np.column_stack((sequence, np.ones(sequence.shape[0])))
+    (a, b, c) = mat.shape
+    mat_rshp = mat.reshape((a, b*c))
+    mat_rshp_reg = np.linalg.lstsq(A, mat_rshp, rcond=None)[0][0]
+    mat_reg = mat_rshp_reg.reshape((b, c))
+    pvalue = f_regression(mat_rshp, sequence)[1].reshape(b, c)
+    return mat_reg, pvalue
+
+
+def convert_longitude_range(data: Union[xr.DataArray, xr.Dataset],
+                            lon: str = 'lon',
+                            center_on_180: bool = True
+                            ) -> Union[xr.DataArray, xr.Dataset]:
+    '''
+    Wrap longitude coordinates of DataArray or Dataset to either -180..179 or 0..359.
+
+    Parameters
+    ----------
+    data : xr.DataArray or xr.Dataset
+        An xarray DataArray or Dataset object containing longitude coordinates.
+    lon : str, optional
+        The name of the longitude coordinate, default is 'lon'.
+    center_on_180 : bool, optional
+        If True, wrap longitude from 0..359 to -180..179;
+        If False, wrap longitude from -180..179 to 0..359.
+
+    Returns
+    -------
+    xr.DataArray or xr.Dataset
+        The DataArray or Dataset with wrapped longitude coordinates.
+    '''
+    # Wrap -180..179 to 0..359
+    if center_on_180:
+        data = data.assign_coords(**{lon: (lambda x: (x[lon] % 360))})
+    # Wrap 0..359 to -180..179
+    else:
+        data = data.assign_coords(
+            **{lon: (lambda x: ((x[lon] + 180) % 360) - 180)})
+    return data.sortby(lon, ascending=True)

skyborn-0.1.0/src/skyborn/gradients.py

@@ -0,0 +1,189 @@
+import numpy as np
+from typing import Union, Optional
+
+EARTH_RADIUS = 6371e3  # 地球平均半径 (m)
+
+
+def calculate_gradient(field: np.ndarray, coordinates: np.ndarray,
+                       axis: int = -1, radius: float = 6371000.0) -> np.ndarray:
+    """计算任意维度数组沿指定坐标的梯度
+
+    Args:
+        field: 待计算梯度的数据场，可以是任意维度的数组，如(time, level, lat, lon)
+        coordinates: 沿着计算梯度的坐标数组，如纬度或经度值
+        axis: 指定计算梯度的维度轴，默认为最后一个维度(-1)
+        radius: 地球半径，默认为6371000.0米，用于纬度梯度计算
+
+    Returns:
+        与输入场相同形状的梯度场
+    """
+    # 检查输入数据维度是否匹配
+    if coordinates.size != field.shape[axis]:
+        raise ValueError(
+            f"坐标数组大小({coordinates.size})与场数据在指定轴上的大小({field.shape[axis]})不匹配")
+
+    # 确定是经度还是纬度坐标
+    is_latitude = False
+    if np.min(coordinates) >= -90 and np.max(coordinates) <= 90:
+        is_latitude = True
+        # 对于纬度，计算实际距离（单位：米）
+        if is_latitude:
+            # 将纬度转换为实际距离
+            distances = coordinates * np.pi / 180.0 * radius
+        else:
+            # 对于经度，我们需要考虑纬度的影响，但这需要额外的纬度信息
+            # 这里简单处理为直接使用经度差
+            distances = coordinates
+
+    # 创建与输入相同形状的输出数组
+    gradient = np.zeros_like(field, dtype=float)
+
+    # 为了使用numpy的高级索引，我们需要创建索引数组
+    ndim = field.ndim
+    idx_ranges = [slice(None)] * ndim
+
+    # 对内部点使用中心差分
+    inner_range = slice(1, field.shape[axis]-1)
+    idx_forward = idx_ranges.copy()
+    idx_forward[axis] = slice(2, field.shape[axis])
+
+    idx_center = idx_ranges.copy()
+    idx_center[axis] = inner_range
+
+    idx_backward = idx_ranges.copy()
+    idx_backward[axis] = slice(0, field.shape[axis]-2)
+
+    # 使用矢量化操作计算内部点的梯度
+    forward_dists = np.diff(distances[1:])
+    backward_dists = np.diff(distances[:-1])
+    total_dists = distances[2:] - distances[:-2]
+
+    # 创建系数数组，形状适合广播
+    shape = [1] * ndim
+    shape[axis] = len(forward_dists)
+
+    a0 = forward_dists.reshape(shape)
+    b0 = backward_dists.reshape(shape)
+    c0 = total_dists.reshape(shape)
+
+    # 使用加权差分公式计算梯度
+    gradient[tuple(idx_center)] = (
+        b0 / a0 / c0 * field[tuple(idx_forward)] -
+        a0 / b0 / c0 * field[tuple(idx_backward)] +
+        (a0 - b0) / a0 / b0 * field[tuple(idx_center)]
+    )
+
+    # 处理边界点（前向和后向差分）
+    # 左边界
+    left_idx = idx_ranges.copy()
+    left_idx[axis] = 0
+    left_idx_plus = idx_ranges.copy()
+    left_idx_plus[axis] = 1
+    gradient[tuple(left_idx)] = (field[tuple(left_idx_plus)] -
+                                 field[tuple(left_idx)]) / (distances[1] - distances[0])
+
+    # 右边界
+    right_idx = idx_ranges.copy()
+    right_idx[axis] = -1
+    right_idx_minus = idx_ranges.copy()
+    right_idx_minus[axis] = -2
+    gradient[tuple(right_idx)] = (field[tuple(right_idx)] -
+                                  field[tuple(right_idx_minus)]) / (distances[-1] - distances[-2])
+
+    return gradient
+
+
+def calculate_meridional_gradient(field: np.ndarray, latitudes: np.ndarray,
+                                  lat_axis: int = -1, radius: float = 6371000.0) -> np.ndarray:
+    """计算经向梯度（沿纬度方向的梯度）
+
+    Args:
+        field: 待计算梯度的数据场，可以是任意维度的数组
+        latitudes: 纬度数组（度）
+        lat_axis: 指定纬度所在的轴，默认为最后一个维度(-1)
+        radius: 地球半径，默认为6371000.0米
+
+    Returns:
+        经向梯度场
+    """
+    return calculate_gradient(field, latitudes, axis=lat_axis, radius=radius)
+
+
+def calculate_vertical_gradient(field: np.ndarray,
+                                pressure: np.ndarray,
+                                pressure_axis: int = -3) -> np.ndarray:
+    """计算垂直梯度（沿气压方向的梯度）
+
+    Args:
+        field: 待计算梯度的数据场
+        pressure: 气压数组（Pa），必须为单调递减
+        pressure_axis: 指定气压所在的轴，默认为倒数第三个维度(-3)
+
+    Returns:
+        垂直梯度场
+    """
+    return calculate_gradient(field, pressure, axis=pressure_axis, radius=None)
+
+
+def calculate_zonal_gradient(field: np.ndarray, longitudes: np.ndarray, latitudes: np.ndarray,
+                             lon_axis: int = -1, lat_axis: int = -2, radius: float = 6371000.0) -> np.ndarray:
+    """计算纬向梯度（沿经度方向的梯度）
+
+    Args:
+        field: 待计算梯度的数据场，可以是任意维度的数组
+        longitudes: 经度数组（度）
+        latitudes: 纬度数组（度），用于计算不同纬度下经度的实际距离
+        lon_axis: 指定经度所在的轴，默认为最后一个维度(-1)
+        lat_axis: 指定纬度所在的轴，默认为倒数第二个维度(-2)
+        radius: 地球半径，默认为6371000.0米
+
+    Returns:
+        纬向梯度场
+    """
+    # 获取纬度因子，用于调整不同纬度下经度间的实际距离
+    cos_lat = np.cos(np.radians(latitudes))
+
+    # 如果场是4D (time, level, lat, lon)
+    if field.ndim == 4 and lon_axis == -1 and lat_axis == -2:
+        # 创建一个广播形状的纬度因子数组
+        cos_lat_expanded = cos_lat.reshape(1, 1, -1, 1)
+
+        # 将经度转换为考虑纬度的实际距离
+        effective_distances = np.radians(
+            longitudes) * radius * cos_lat_expanded
+
+        # 现在计算梯度
+        return calculate_gradient(field, effective_distances, axis=lon_axis, radius=1.0)
+
+    # 如果场是3D (time, lat, lon)
+    elif field.ndim == 3 and lon_axis == -1 and lat_axis == -2:
+        cos_lat_expanded = cos_lat.reshape(1, -1, 1)
+        effective_distances = np.radians(
+            longitudes) * radius * cos_lat_expanded
+        return calculate_gradient(field, effective_distances, axis=lon_axis, radius=1.0)
+
+    else:
+        # 对于其他维度组合，需要创建适当的广播形状
+        broadcast_shape = [1] * field.ndim
+        broadcast_shape[lat_axis] = len(latitudes)
+        cos_lat_expanded = cos_lat.reshape(broadcast_shape)
+
+        # 创建有效距离数组
+        effective_longitudes = np.radians(longitudes) * radius
+
+        # 对于每个纬度，计算梯度
+        result = np.zeros_like(field)
+
+        # 循环处理每个纬度（这部分实现取决于具体数据结构，可能需要调整）
+        for i in range(len(latitudes)):
+            idx = [slice(None)] * field.ndim
+            idx[lat_axis] = i
+
+            # 调整当前纬度的经度距离
+            current_effective_dist = effective_longitudes * cos_lat[i]
+
+            # 计算当前纬度的梯度
+            result[tuple(idx)] = calculate_gradient(
+                field[tuple(idx)], current_effective_dist, axis=lon_axis, radius=1.0)
+
+        return result

skyborn-0.1.0/src/skyborn/io.py

@@ -0,0 +1,25 @@
+import xarray as xr
+import cfgrib
+
+def read_netcdf(file_path: str) -> xr.Dataset:
+    """读取NetCDF格式气象数据
+    
+    Args:
+        file_path: 文件路径
+        
+    Returns:
+        xarray Dataset对象
+    """
+    return xr.open_dataset(file_path)
+
+def read_grib(file_path: str, **kwargs) -> xr.Dataset:
+    """读取GRIB格式气象数据
+    
+    Args:
+        file_path: 文件路径
+        kwargs: 传递给cfgrib.open_dataset的额外参数
+        
+    Returns:
+        xarray Dataset对象
+    """
+    return cfgrib.open_dataset(file_path, **kwargs)

skyborn-0.1.0/src/skyborn/plotting.py

@@ -0,0 +1,67 @@
+import matplotlib.pyplot as plt
+import matplotlib.transforms as mtransforms
+import numpy as np
+
+def add_equal_axes(ax, loc, pad, width):
+    '''
+    在原有的Axes旁新添一个等高或等宽的Axes并返回该对象.
+
+    Parameters
+    ----------
+    ax : Axes or array_like of Axes
+        原有的Axes,也可以为一组Axes构成的数组.
+
+    loc : {'left', 'right', 'bottom', 'top'}
+        新Axes相对于旧Axes的位置.
+
+    pad : float
+        新Axes与旧Axes的间距.
+
+    width: float
+        当loc='left'或'right'时,width表示新Axes的宽度.
+        当loc='bottom'或'top'时,width表示新Axes的高度.
+
+    Returns
+    -------
+    ax_new : Axes
+        新Axes对象.
+    '''
+    # 无论ax是单个还是一组Axes,获取ax的大小位置.
+    axes = np.atleast_1d(ax).ravel()
+    bbox = mtransforms.Bbox.union([ax.get_position() for ax in axes])
+
+    # 决定新Axes的大小位置.
+    if loc == 'left':
+        x0_new = bbox.x0 - pad - width
+        x1_new = x0_new + width
+        y0_new = bbox.y0
+        y1_new = bbox.y1
+    elif loc == 'right':
+        x0_new = bbox.x1 + pad
+        x1_new = x0_new + width
+        y0_new = bbox.y0
+        y1_new = bbox.y1
+    elif loc == 'bottom':
+        x0_new = bbox.x0
+        x1_new = bbox.x1
+        y0_new = bbox.y0 - pad - width
+        y1_new = y0_new + width
+    elif loc == 'top':
+        x0_new = bbox.x0
+        x1_new = bbox.x1
+        y0_new = bbox.y1 + pad
+        y1_new = y0_new + width
+
+    # 创建新Axes.
+    fig = axes[0].get_figure()
+    bbox_new = mtransforms.Bbox.from_extents(x0_new, y0_new, x1_new, y1_new)
+    ax_new = fig.add_axes(bbox_new)
+
+    return ax_new
+
+def createFigure(figsize=(12, 8), dpi=300, subplotAdj=None, **kwargs):
+    figsize = figsize
+    figure = plt.figure(figsize=figsize, dpi=dpi, **kwargs)
+    if subplotAdj is not None:
+        plt.subplots_adjust(**subplotAdj)
+    return figure

skyborn-0.1.0/src/skyborn.egg-info/PKG-INFO

@@ -0,0 +1,28 @@
+Metadata-Version: 2.2
+Name: skyborn
+Version: 0.1.0
+Summary: Atmospheric science research utilities
+Author: Qianye Su
+Author-email: suqianye2000@gmail.com
+License: MIT
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Dist: numpy>=1.20.0
+Requires-Dist: xarray>=0.19.0
+Requires-Dist: matplotlib>=3.4.0
+Requires-Dist: cartopy>=0.20.0
+Requires-Dist: netCDF4>=1.5.7
+Requires-Dist: metpy>=1.1.0
+Requires-Dist: cfgrib>=0.9.9
+Requires-Dist: eccodes>=1.4.0
+Requires-Dist: scikit-learn>=1.0.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: summary
+
+Skyborn is a tool for easy plotting ERA5 weather data.

skyborn-0.1.0/src/skyborn.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+README.md
+setup.py
+src/skyborn/__init__.py
+src/skyborn/calculations.py
+src/skyborn/gradients.py
+src/skyborn/io.py
+src/skyborn/plotting.py
+src/skyborn.egg-info/PKG-INFO
+src/skyborn.egg-info/SOURCES.txt
+src/skyborn.egg-info/dependency_links.txt
+src/skyborn.egg-info/requires.txt
+src/skyborn.egg-info/top_level.txt

skyborn-0.1.0/src/skyborn.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

skyborn-0.1.0/src/skyborn.egg-info/requires.txt

@@ -0,0 +1,9 @@
+numpy>=1.20.0
+xarray>=0.19.0
+matplotlib>=3.4.0
+cartopy>=0.20.0
+netCDF4>=1.5.7
+metpy>=1.1.0
+cfgrib>=0.9.9
+eccodes>=1.4.0
+scikit-learn>=1.0.0

skyborn-0.1.0/src/skyborn.egg-info/top_level.txt

@@ -0,0 +1 @@
+skyborn