rsnumpy 1.0.1__tar.gz → 1.0.2__tar.gz

{rsnumpy-1.0.1 → rsnumpy-1.0.2}/.gitignore

@@ -21,7 +21,6 @@ uv.lock
 # IDE
 .vscode/
 .idea/
-.trae/
 *.swp
 *.swo
 *~
@@ -46,3 +45,6 @@ check*.py
 .venv
 .env
 .env.local
+.trae/
+.trae/rules/git-commit-message.md
+git-commit-message.md

{rsnumpy-1.0.1 → rsnumpy-1.0.2}/Cargo.lock

@@ -62,6 +62,15 @@ dependencies = [
  "libbz2-rs-sys",
 ]
 
+[[package]]
+name = "cblas-sys"
+version = "0.1.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b6feecd82cce51b0204cf063f0041d69f24ce83f680d87514b004248e7b0fa65"
+dependencies = [
+ "libc",
+]
+
 [[package]]
 name = "cc"
 version = "1.2.64"
@@ -461,6 +470,8 @@ version = "0.17.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "520080814a7a6b4a6e9070823bb24b4531daac8c4627e08ba5de8c5ef2f2752d"
 dependencies = [
+ "cblas-sys",
+ "libc",
  "matrixmultiply",
  "num-complex",
  "num-integer",
@@ -469,6 +480,7 @@ dependencies = [
  "portable-atomic-util",
  "rawpointer",
  "rayon",
+ "serde",
 ]
 
 [[package]]
@@ -721,15 +733,17 @@ dependencies = [
 
 [[package]]
 name = "rsnumpy"
-version = "1.0.1"
+version = "1.0.2"
 dependencies = [
  "ndarray",
+ "num-complex",
  "num-traits",
  "pyo3",
  "rand",
  "rand_distr",
  "rayon",
  "rustfft",
+ "thiserror",
  "zip",
 ]
 
@@ -864,6 +878,26 @@ version = "0.13.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "adb6935a6f5c20170eeceb1a3835a49e12e19d792f6dd344ccc76a985ca5a6ca"
 
+[[package]]
+name = "thiserror"
+version = "2.0.18"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4288b5bcbc7920c07a1149a35cf9590a2aa808e0bc1eafaade0b80947865fbc4"
+dependencies = [
+ "thiserror-impl",
+]
+
+[[package]]
+name = "thiserror-impl"
+version = "2.0.18"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ebc4ee7f67670e9b64d05fa4253e753e016c6c95ff35b89b7941d6b856dec1d5"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+]
+
 [[package]]
 name = "time"
 version = "0.3.49"

rsnumpy-1.0.2/Cargo.toml

@@ -0,0 +1,36 @@
+[package]
+name = "rsnumpy"
+version = "1.0.2"
+edition = "2024"
+readme = "README.md"
+
+[lib]
+name = "rsnumpy"
+crate-type = ["cdylib"]
+
+[dependencies]
+pyo3 = { version = "0.29.0", features = ["extension-module", "abi3-py38"] }
+ndarray = { version = "0.17.2", features = ["rayon", "blas", "serde"] }
+rayon = "1.12.0"
+rand = "0.10.1"
+rand_distr = "0.6.0"
+num-traits = "0.2.19"
+rustfft = "6.4.1"
+zip = "8.6.0"
+num-complex = "0.4.6"
+thiserror = "2.0.0"
+
+[features]
+default = []
+blas = ["ndarray/blas"]
+
+[profile.release]
+opt-level = 3
+lto = "fat"
+codegen-units = 1
+panic = "abort"
+strip = true
+
+[profile.bench]
+opt-level = 3
+debug = false

{rsnumpy-1.0.1 → rsnumpy-1.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rsnumpy
-Version: 1.0.1
+Version: 1.0.2
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
@@ -15,7 +15,6 @@ Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
 
 **A Rust-powered NumPy-compatible array library.**
 
-![Build Status](https://github.com/your-username/rsnumpy/actions/workflows/ci.yml/badge.svg)
 ![License](https://img.shields.io/badge/license-MIT-blue.svg)
 ![Python Version](https://img.shields.io/badge/python-%3E%3D%203.10-blue.svg)
 ![Rust Version](https://img.shields.io/badge/rust-%3E%3D%201.75-orange.svg)

{rsnumpy-1.0.1 → rsnumpy-1.0.2}/README.md

@@ -2,7 +2,6 @@
 
 **A Rust-powered NumPy-compatible array library.**
 
-![Build Status](https://github.com/your-username/rsnumpy/actions/workflows/ci.yml/badge.svg)
 ![License](https://img.shields.io/badge/license-MIT-blue.svg)
 ![Python Version](https://img.shields.io/badge/python-%3E%3D%203.10-blue.svg)
 ![Rust Version](https://img.shields.io/badge/rust-%3E%3D%201.75-orange.svg)

rsnumpy-1.0.2/RELEASE_NOTES.md

@@ -0,0 +1,82 @@
+# Release Notes
+
+## v1.0.2 (2026-06-24)
+
+### 性能优化
+
+- **索引模块优化**：向量预分配，提升花式索引性能
+- **数组拼接优化**：`concatenate`、`stack` 等函数性能提升
+- **数组格式化输出优化**：预分配字符串缓冲区、简化切片遍历逻辑
+- **工具函数优化**：`arange`、`reshape` 等函数的预分配和边界处理
+- **IO 函数优化**：`loadtxt`、`savez` 等函数的预分配和遍历逻辑
+
+### 模块重构
+
+- **随机数模块重构**：使用 const 初始化线程局部 RNG、优化形状解析预分配、使用 `div_ceil` 计算分块大小、简化采样类型转换
+- **线性代数模块重构**：将 Python 绑定函数移至 PyO3 detach 闭包，避免 GIL 持有过长时间
+- **统计函数优化**：简化 `polyfit`、`cov` 等计算的循环逻辑
+
+### Bug 修复
+
+- 修复 `median`、`percentile` 等统计函数的偶数长度判断逻辑
+- 修复部分边界检查的冗余代码
+
+### 构建优化
+
+- 新增 `build_wheel.sh` 自动同步版本号的逻辑
+- 优化构建配置，二进制大小略有优化
+
+---
+
+## v1.0.1 (2026-06-24)
+
+### 功能增强
+
+- **新增 ArrayFlags 支持**：提供数组标志位查询功能
+- **新增 like-series 数组创建函数**：支持根据已有数组的属性创建新数组
+- **改进 logspace 函数**：优化对数空间生成算法
+
+### 性能提升
+
+- **并行加速矩阵运算**：利用多线程并行化矩阵运算
+- **随机数生成优化**：并行加速随机数生成
+- **构建配置优化**：调整编译参数，提升构建效率
+
+### 代码重构
+
+- **迁移矩阵和数组操作到 Rust 后端**：将大量矩阵和数组操作从 Python 层迁移到 Rust 层实现
+- **重构 matlib.py 的矩阵函数实现**：优化矩阵函数的代码结构
+
+### 文档更新
+
+- 更新 README 中 Python 版本支持要求至 3.10+
+- 更新 README 文档细节并修复格式问题
+
+---
+
+## v1.0.0 (Initial Release)
+
+### 核心功能
+
+- **高性能多维数组**：基于 Rust 的 ndarray 实现，提供 NumPy 兼容的 API
+- **数学函数**：支持 sin、cos、exp、sqrt、log 等常见数学函数
+- **数组操作**：支持 reshape、transpose、切片、拼接等操作
+- **统计函数**：支持 sum、mean、std、var、max、min、argmin、argmax 等
+- **线性代数**：支持 det、inv、norm、solve、svd、qr 等矩阵运算
+- **FFT**：支持快速傅里叶变换及其逆变换
+- **随机数**：支持 uniform、normal、integers 等分布，兼容新旧两种 API
+- **多项式**：支持多项式求值、求根、运算和拟合
+- **文件 I/O**：支持 .npy、.npz 和文本文件的读写
+
+### 优势
+
+- **性能**：核心计算使用 Rust 实现，部分操作（sum、mean、dot 等）比 NumPy 快 1.2x-2.0x
+- **类型安全**：利用 Rust 的强类型系统避免运行时错误
+- **内存安全**：无数据竞争，无内存泄漏
+- **完整 API**：覆盖 NumPy 常用功能
+
+### 支持平台
+
+- Python ≥ 3.10
+- Rust ≥ 1.75
+- macOS、Linux、Windows

{rsnumpy-1.0.1 → rsnumpy-1.0.2}/build_wheel.sh

@@ -4,17 +4,27 @@ set -euo pipefail
 # Build Rust lib into Python wheel using maturin
 # Usage: scripts/build_wheel.sh [--release|--debug] [--out-dir DIR] [--python PYTHON_EXEC]
 
+# ========== 从 Cargo.toml 读取元信息 ==========
+NAME=$(grep -m1 '^name = ' Cargo.toml | sed 's/name = "\(.*\)"/\1/')
+VERSION=$(grep -m1 '^version = ' Cargo.toml | sed 's/version = "\(.*\)"/\1/')
+
+echo "Project: $NAME, Version: $VERSION"
+
+# ========== 同步版本到 pyproject.toml ==========
+sed -i '' "s/^version = \".*\"/version = \"$VERSION\"/" pyproject.toml
+echo "  -> pyproject.toml version updated to $VERSION"
+
+# ========== 同步版本到 python/rsnumpy/__init__.py ==========
+sed -i '' "s/^__version__ = \".*\"/__version__ = \"$VERSION\"/" python/rsnumpy/__init__.py
+echo "  -> python/rsnumpy/__init__.py __version__ updated to $VERSION"
+
 RELEASE=true
 OUT_DIR="wheelhouse"
-cache_file="/Users/user/Desktop/rust_project/rsplotlib/python/rsplotlib/rsplotlib.cpython-313-darwin.so"
+
 # 如果 wheelhouse 目录存在, 则删除
 if [[ -d "$OUT_DIR" ]]; then
   rm -rf "$OUT_DIR"
 fi
-# 如果 cache_file 目录存在, 则删除
-if [[ -f "$cache_file" ]]; then
-  rm "$cache_file"
-fi
 
 # Default python executable; may be overridden by --python or positional arg
 PYTHON_EXEC="python"
@@ -80,4 +90,4 @@ else
   echo "Warning: no .whl file found in $OUT_DIR; skip install." >&2
 fi
 
-echo "Done."
+echo "Done."

{rsnumpy-1.0.1 → rsnumpy-1.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "rsnumpy"
-version = "1.0.1"
+version = "1.0.2"
 requires-python = ">=3.10"
 description = "A Rust-powered NumPy-compatible array library."
 readme = "README.md"

{rsnumpy-1.0.1 → rsnumpy-1.0.2}/python/rsnumpy/__init__.py

@@ -29,7 +29,7 @@ from .random import random_module as _random_module
 from . import char as _char_module
 from . import matlib as _matlib_module
 
-__version__ = "1.0.1"
+__version__ = "1.0.2"
 
 class ArrayFlags:
     """数组内存布局信息，与 NumPy 的 np.ndarray.flags 兼容。"""
@@ -1338,11 +1338,7 @@ def repmat(a, m, n):
     a = _to_ndarray(a)
     if a.ndim == 1:
         a = a.reshape(1, a.size)
-    result = []
-    for _ in range(m):
-        for row in range(a.shape[0]):
-            result.append(a[row].tolist() * n)
-    return asarray(result)
+    return tile(a, (m, n))
 
 
 def _to_ndarray(obj):
@@ -1468,28 +1464,28 @@ def zeros_like(a, dtype=None, order='K', subok=True, shape=None):
     """返回与输入形状相同的零数组。"""
     arr = ndarray(a)
     _dtype = dtype if dtype is not None else getattr(arr, '_dtype', 'float64')
-    return zeros(arr.shape, dtype=_dtype)
+    return ndarray(_core.zeros_like(arr._array), _dtype=_dtype)
 
 
 def ones_like(a, dtype=None, order='K', subok=True, shape=None):
     """返回与输入形状相同的1数组。"""
     arr = ndarray(a)
     _dtype = dtype if dtype is not None else getattr(arr, '_dtype', 'float64')
-    return ones(arr.shape, dtype=_dtype)
+    return ndarray(_core.ones_like(arr._array), _dtype=_dtype)
 
 
 def empty_like(a, dtype=None, order='K', subok=True, shape=None):
     """返回与输入形状相同的空数组。"""
     arr = ndarray(a)
     _dtype = dtype if dtype is not None else getattr(arr, '_dtype', 'float64')
-    return empty(arr.shape, dtype=_dtype)
+    return ndarray(_core.empty_like(arr._array), _dtype=_dtype)
 
 
 def full_like(a, fill_value, dtype=None, order='K', subok=True, shape=None):
     """返回与输入形状相同的填充数组。"""
     arr = ndarray(a)
     _dtype = dtype if dtype is not None else getattr(arr, '_dtype', 'float64')
-    return full(arr.shape, fill_value, dtype=_dtype)
+    return ndarray(_core.full_like(arr._array, float(fill_value)), _dtype=_dtype)
 
 
 def eye(N, M=None, k=0, dtype=None, order='C'):
@@ -1517,32 +1513,25 @@ def linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis
     start_val = _scalar(_ensure(start))
     stop_val = _scalar(_ensure(stop))
     _dtype = _resolve_dtype(dtype)
-    if num == 1:
-        result = ndarray([start_val], _dtype=_dtype)
-        step = 0.0
-    else:
-        result = ndarray(_core.linspace(start_val, stop_val, num, endpoint=endpoint))
-        if _dtype != "float64":
-            result._dtype = _dtype
-        step = (stop_val - start_val) / (num - 1) if endpoint else (stop_val - start_val) / num
+    result = ndarray(_core.linspace(start_val, stop_val, num, endpoint=endpoint), _dtype=_dtype)
     if retstep:
+        step = (stop_val - start_val) / (num - 1) if endpoint else (stop_val - start_val) / num
         return result, _float64(step)
     return result
 
 
 def logspace(start, stop, num=50, endpoint=True, base=10.0, dtype=None, axis=0):
     """返回对数刻度上均匀间隔的数字。"""
-    return power(ndarray([float(base)]), linspace(start, stop, num, endpoint))
+    _dtype = _resolve_dtype(dtype) if dtype else "float64"
+    return ndarray(_core.logspace(float(start), float(stop), num, float(base)), _dtype=_dtype)
 
 
 def geomspace(start, stop, num=50, endpoint=True, dtype=None, axis=0):
     """返回几何级数上均匀间隔的数字。"""
-    import math
     if start <= 0 or stop <= 0:
         raise ValueError("geomspace requires positive start and stop values")
-    log_start = math.log(start)
-    log_stop = math.log(stop)
-    return exp(linspace(log_start, log_stop, num, endpoint))
+    _dtype = _resolve_dtype(dtype) if dtype else "float64"
+    return ndarray(_core.geomspace(float(start), float(stop), num), _dtype=_dtype)
 
 
 # ========== 索引函数 ==========
@@ -1943,12 +1932,7 @@ def isnan(x):
 
 def binary_repr(num, width=None):
     """返回整数的二进制表示字符串。"""
-    if width is None:
-        width = num.bit_length() or 1
-    if num >= 0:
-        return format(num, 'b').zfill(width)
-    # 负数：补码表示
-    return format((1 << width) + num, 'b').zfill(width)
+    return _core.binary_repr(num, width)
 
 
 def isinf(x):

rsnumpy-1.0.2/python/rsnumpy/array_methods.py

@@ -0,0 +1,225 @@
+"""
+ndarray 对象方法模块
+
+此模块包含 rsnumpy.ndarray 类的所有对象方法和属性。
+"""
+
+import rsnumpy._core as _core
+from .__init__ import ndarray, _ensure
+
+
+def _wrap_result(result, dtype="float64"):
+    """包装结果为 ndarray 对象"""
+    if hasattr(result, '__class__') and result.__class__.__name__ == 'ndarray':
+        return ndarray._wrap(result, _dtype=dtype)
+    return result
+
+
+class NdArrayMethods:
+    """ndarray 对象方法集合"""
+    
+    @staticmethod
+    def astype(arr, dtype):
+        """将数组转换为指定的数据类型。"""
+        return _wrap_result(arr._array.astype(dtype))
+    
+    @staticmethod
+    def reshape(arr, *shape):
+        """改变数组形状而不改变数据。"""
+        if len(shape) == 1 and isinstance(shape[0], (tuple, list)):
+            shape = shape[0]
+        else:
+            shape = shape
+        result = arr._array.reshape(shape)
+        return ndarray._wrap(result, _dtype=getattr(arr, '_dtype', "float64"), _fields=getattr(arr, '_fields', None), _raw_data=getattr(arr, '_raw_data', None))
+    
+    @staticmethod
+    def resize(arr, new_shape):
+        """改变数组形状和大小。"""
+        if isinstance(new_shape, int):
+            new_shape = (new_shape,)
+        elif isinstance(new_shape, list):
+            new_shape = tuple(new_shape)
+        arr._array.resize(new_shape)
+    
+    @staticmethod
+    def ravel(arr, order='C'):
+        """将数组展平为一维。"""
+        if order == 'A' or order == 'K':
+            order = 'C'
+        return NdArrayMethods.flatten(arr, order)
+    
+    @staticmethod
+    def flatten(arr, order='C'):
+        """展平数组为一维。"""
+        dtype = getattr(arr, '_dtype', "float64")
+        result = _core.flatten_full(arr._array, order)
+        return _wrap_result(result, dtype)
+    
+    @staticmethod
+    def transpose(arr, *axes):
+        """转置数组。"""
+        dtype = getattr(arr, '_dtype', "float64")
+        fields = getattr(arr, '_fields', None)
+        raw = getattr(arr, '_raw_data', None)
+        if len(axes) == 0:
+            result = arr._array.T
+        else:
+            result = _core.transpose(arr._array)
+        return ndarray._wrap(result, _dtype=dtype, _fields=fields, _raw_data=raw)
+    
+    @staticmethod
+    def swapaxes(arr, axis1, axis2):
+        """交换两个轴的位置。"""
+        dtype = getattr(arr, '_dtype', "float64")
+        fields = getattr(arr, '_fields', None)
+        raw_data = getattr(arr, '_raw_data', None)
+        result = _core.swapaxes(arr._array, axis1, axis2)
+        return ndarray._wrap(result, _dtype=dtype, _fields=fields, _raw_data=raw_data)
+    
+    @staticmethod
+    def squeeze(arr, axis=None):
+        """移除长度为1的轴。"""
+        return _wrap_result(arr._array.squeeze())
+    
+    @staticmethod
+    def max(arr, axis=None):
+        """返回数组的最大值。"""
+        result = arr._array.max(axis)
+        if axis is None and hasattr(result, 'tolist'):
+            return result.tolist()
+        return _wrap_result(result)
+    
+    @staticmethod
+    def min(arr, axis=None):
+        """返回数组的最小值。"""
+        result = arr._array.min(axis)
+        if axis is None and hasattr(result, 'tolist'):
+            return result.tolist()
+        return _wrap_result(result)
+    
+    @staticmethod
+    def mean(arr, axis=None):
+        """计算数组的平均值。"""
+        result = arr._array.mean(axis)
+        if axis is None and hasattr(result, 'tolist'):
+            return result.tolist()
+        return _wrap_result(result)
+    
+    @staticmethod
+    def std(arr, axis=None):
+        """计算数组的标准差。"""
+        result = arr._array.std(axis)
+        if axis is None and hasattr(result, 'tolist'):
+            return result.tolist()
+        return _wrap_result(result)
+    
+    @staticmethod
+    def var(arr, axis=None):
+        """计算数组的方差。"""
+        result = arr._array.var(axis)
+        if axis is None and hasattr(result, 'tolist'):
+            return result.tolist()
+        return _wrap_result(result)
+    
+    @staticmethod
+    def sum(arr, axis=None):
+        """计算数组元素的和。"""
+        result = arr._array.sum(axis)
+        if axis is None and hasattr(result, 'tolist'):
+            return result.tolist()
+        return _wrap_result(result)
+    
+    @staticmethod
+    def prod(arr, axis=None):
+        """计算数组元素的乘积。"""
+        result = arr._array.prod(axis)
+        if axis is None and hasattr(result, 'tolist'):
+            return result.tolist()
+        return _wrap_result(result)
+    
+    @staticmethod
+    def cumsum(arr, axis=None):
+        """计算累积和。"""
+        result = arr._array.cumsum(axis)
+        return _wrap_result(result)
+    
+    @staticmethod
+    def cumprod(arr, axis=None):
+        """计算累积乘积。"""
+        result = arr._array.cumprod(axis)
+        return _wrap_result(result)
+    
+    @staticmethod
+    def argmax(arr, axis=None):
+        """返回最大值的索引。"""
+        return _wrap_result(_core.argmax_axis(arr._array, axis))
+    
+    @staticmethod
+    def argmin(arr, axis=None):
+        """返回最小值的索引。"""
+        return _wrap_result(_core.argmin_axis(arr._array, axis))
+    
+    @staticmethod
+    def argsort(arr, axis=-1):
+        """返回排序后的索引。"""
+        result = arr._array.argsort(axis)
+        return _wrap_result(result)
+    
+    @staticmethod
+    def sort(arr, axis=-1):
+        """原地排序数组。"""
+        result = arr._array.sort(axis)
+        if hasattr(result, 'shape') and result.shape == arr.shape:
+            arr._array = result
+        else:
+            arr._array.data = result.data
+    
+    @staticmethod
+    def diagonal(arr, offset=0, axis1=0, axis2=1):
+        """返回数组的对角线元素。"""
+        result = arr._array.diagonal(offset, axis1, axis2)
+        return _wrap_result(result)
+    
+    @staticmethod
+    def trace(arr, offset=0, axis1=0, axis2=1):
+        """计算数组的迹（对角线元素之和）。"""
+        return arr._array.trace(offset, axis1, axis2)
+    
+    @staticmethod
+    def fill(arr, value):
+        """用指定值填充数组。"""
+        arr._array.fill(value)
+    
+    @staticmethod
+    def item(arr, *args):
+        """获取数组的单个元素。"""
+        return arr._array.item(*args)
+    
+    @staticmethod
+    def tolist(arr):
+        """将数组转换为 Python 列表。"""
+        return arr._array.tolist()
+    
+    @staticmethod
+    def take(arr, indices, axis=None):
+        """根据索引获取元素。"""
+        result = arr._array.take(_ensure(indices), axis)
+        return _wrap_result(result)
+    
+    @staticmethod
+    def put(arr, indices, values):
+        """将值放入数组的指定位置。"""
+        arr._array.put(_ensure(indices), _ensure(values))
+    
+    @staticmethod
+    def repeat(arr, repeats, axis=None):
+        """重复数组元素。"""
+        result = arr._array.repeat(repeats, axis)
+        return _wrap_result(result)
+    
+    @staticmethod
+    def nonzero(arr):
+        """返回非零元素的索引。"""
+        raw = _core.nonzero_arrs(arr._array)
+        return tuple(ndarray._wrap(r) for r in raw)