rsnumpy 1.0.6__tar.gz → 1.1.3__tar.gz

{rsnumpy-1.0.6 → rsnumpy-1.1.3}/.gitignore

@@ -1,5 +1,6 @@
 # Rust
 target/
+_split_tool/
 wheelhouse/
 Cargo.lock
 rsnum*.whl
@@ -41,7 +42,9 @@ debug*.py
 check*.py
 test2*.py
 map*.jpeg
+bench_extra.py
 *.png
+runtest_matplotlib.py
 
 # Environment
 .venv

{rsnumpy-1.0.6 → rsnumpy-1.1.3}/Cargo.lock

@@ -2,6 +2,12 @@
 # It is not intended for manual editing.
 version = 4
 
+[[package]]
+name = "accelerate-src"
+version = "0.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "415ed64958754dbe991900f3940677e6a7eefb4d7367afd70d642677b0c7d19d"
+
 [[package]]
 name = "adler2"
 version = "2.0.1"
@@ -25,6 +31,15 @@ version = "1.5.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
 
+[[package]]
+name = "blas-src"
+version = "0.10.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b95e83dc868db96e69795c0213143095f03de9dd3252f205d4ac716e4076a7e0"
+dependencies = [
+ "accelerate-src",
+]
+
 [[package]]
 name = "block-buffer"
 version = "0.12.1"
@@ -140,11 +155,20 @@ dependencies = [
  "cfg-if",
 ]
 
+[[package]]
+name = "crossbeam-channel"
+version = "0.5.16"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d85363c37faeca707aef026efa9f3b34d077bce547e48f770770625c6013679e"
+dependencies = [
+ "crossbeam-utils",
+]
+
 [[package]]
 name = "crossbeam-deque"
-version = "0.8.6"
+version = "0.8.7"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "9dd111b7b7f7d55b72c0a6ae361660ee5853c9af73f70c3c2ef6858b950e2e51"
+checksum = "5181e0de7b61eb03a81e347d6dd8797bae9da5146707b51077e2d71a54ec0ceb"
 dependencies = [
  "crossbeam-epoch",
  "crossbeam-utils",
@@ -152,18 +176,18 @@ dependencies = [
 
 [[package]]
 name = "crossbeam-epoch"
-version = "0.9.18"
+version = "0.9.20"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "5b82ac4a3c2ca9c3460964f020e1402edd5753411d7737aa39c3714ad1b5420e"
+checksum = "2d6914041f254d6e9176c01941b21115dcfb7089e55135a35411081bd106ef3f"
 dependencies = [
  "crossbeam-utils",
 ]
 
 [[package]]
 name = "crossbeam-utils"
-version = "0.8.21"
+version = "0.8.22"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d0a5c400df2834b80a4c3327b3aad3a4c4cd4de0629063962b03235697506a28"
+checksum = "61803da095bee82a81bb1a452ecc25d3b2f1416d1897eb86430c6159ef717c17"
 
 [[package]]
 name = "crypto-common"
@@ -286,6 +310,12 @@ version = "0.5.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea"
 
+[[package]]
+name = "hermit-abi"
+version = "0.5.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fc0fef456e4baa96da950455cd02c081ca953b141298e41db3fc7e36b1da849c"
+
 [[package]]
 name = "hmac"
 version = "0.13.0"
@@ -384,14 +414,17 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "a06de3016e9fae57a36fd14dba131fccf49f74b40b7fbdb472f96e361ec71a08"
 dependencies = [
  "autocfg",
+ "num_cpus",
+ "once_cell",
  "rawpointer",
+ "thread-tree",
 ]
 
 [[package]]
 name = "memchr"
-version = "2.8.2"
+version = "2.8.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "88904434abc2901f197fe8cc55f0445e7ded921dba5911dad2e2b39b48e663c4"
+checksum = "cf8baf1c55e62ffcace7a9f06f4bd9cd3f0c4beb022d3b367256b91b87513d98"
 
 [[package]]
 name = "miniz_oxide"
@@ -456,6 +489,16 @@ dependencies = [
  "libm",
 ]
 
+[[package]]
+name = "num_cpus"
+version = "1.17.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "91df4bbde75afed763b708b7eee1e8e7651e02d97f6d5dd763e89367e957b23b"
+dependencies = [
+ "hermit-abi",
+ "libc",
+]
+
 [[package]]
 name = "once_cell"
 version = "1.21.4"
@@ -656,8 +699,9 @@ dependencies = [
 
 [[package]]
 name = "rsnumpy"
-version = "1.0.6"
+version = "1.1.3"
 dependencies = [
+ "blas-src",
  "ndarray",
  "num-complex",
  "num-traits",
@@ -686,9 +730,9 @@ dependencies = [
 
 [[package]]
 name = "rustversion"
-version = "1.0.22"
+version = "1.0.23"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "b39cdef0fa800fc44525c84ccb54a029961a8215f9619753635a9c0d2538d46d"
+checksum = "cf54715a573b99ac80df0bc206da022bcd442c974952c7b9720069370852e21f"
 
 [[package]]
 name = "serde"
@@ -802,6 +846,15 @@ dependencies = [
  "syn",
 ]
 
+[[package]]
+name = "thread-tree"
+version = "0.3.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ffbd370cb847953a25954d9f63e14824a36113f8c72eecf6eccef5dc4b45d630"
+dependencies = [
+ "crossbeam-channel",
+]
+
 [[package]]
 name = "time"
 version = "0.3.53"
@@ -930,9 +983,9 @@ dependencies = [
 
 [[package]]
 name = "zlib-rs"
-version = "0.6.5"
+version = "0.6.6"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "5431d5661c32445236631278f27946e444ddafe4684cac70b185272d4f9c52d5"
+checksum = "b142a20ec14a91d5bc708c1dc21b080c550113d8aa77afa29635673a65dd02c5"
 
 [[package]]
 name = "zopfli"

rsnumpy-1.1.3/Cargo.toml

@@ -0,0 +1,49 @@
+[package]
+name = "rsnumpy"
+version = "1.1.3"
+edition = "2024"
+readme = "README.md"
+
+[lib]
+name = "rsnumpy"
+crate-type = ["cdylib"]
+
+[dependencies]
+pyo3 = { version = "0.29.0", features = ["extension-module"] }
+# matrixmultiply-threading：非 BLAS 平台上让纯 Rust GEMM 内核多线程并行。
+ndarray = { version = "0.17.2", features = [
+    "rayon",
+    "serde",
+    "matrixmultiply-threading",
+] }
+rayon = "1.12.0"
+rand = "0.10.2"
+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"
+
+# macOS：链接系统自带的 Accelerate 框架作为 BLAS 后端（无需额外安装），
+# matmul/dot 由 ndarray 的 blas 特性自动分派到原生高性能 GEMM。
+# 其它平台不启用 BLAS，自动回退到上面的 matrixmultiply 多线程纯 Rust 内核。
+[target.'cfg(target_os = "macos")'.dependencies]
+ndarray = { version = "0.17.2", features = ["blas"] }
+blas-src = { version = "0.10", features = ["accelerate"] }
+
+[profile.release]
+opt-level = 3
+lto = "fat"
+codegen-units = 1
+strip = "symbols"
+
+[profile.dev]
+opt-level = 0
+debug = true
+
+[lints.clippy]
+too_many_arguments = "allow"
+type_complexity = "allow"
+cast_precision_loss = "allow"
+cast_sign_loss = "allow"

{rsnumpy-1.0.6 → rsnumpy-1.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rsnumpy
-Version: 1.0.6
+Version: 1.1.3
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
@@ -51,15 +51,19 @@ Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
 rsnumpy/
 ├── src/                       # Rust 源码
 │   ├── lib.rs                 # 核心 ndarray 与通用函数
+│   ├── indexing.rs            # 多维索引与切片
 │   ├── fft.rs                 # 快速傅里叶变换
 │   ├── linalg.rs              # 线性代数
-│   └── random.rs              # 随机数生成
-├── rsnumpy/                     # Python 薄包装
+│   └── random.rs              # 随机数生成（可复现并行采样）
+├── python/rsnumpy/            # Python 薄包装
 │   ├── __init__.py            # 主模块，整合所有 API
 │   ├── array_methods.py       # ndarray 对象方法
 │   ├── array_ops.py           # 数组操作函数
+│   ├── _extra.py              # 补充 API（别名、nan 系列、集合运算、窗函数等）
 │   ├── math_functions.py      # 数学函数
 │   ├── statistics.py          # 统计函数
+│   ├── char.py                # 字符串数组函数
+│   ├── matlib.py              # 矩阵便捷构造
 │   ├── io.py                  # 文件 I/O
 │   ├── linalg/                # 线性代数子模块
 │   ├── polynomial/            # 多项式子模块
@@ -283,6 +287,8 @@ print(np.random.rand(3).tolist())
 print(np.random.randn(3).tolist())
 ```
 
+> **可复现性**：相同 `seed` 保证生成相同的结果。大数组采样在 Rust 层并行执行，采用固定分块 + `splitmix64` 派生各分块的独立种子，因此**结果与线程数无关**，在不同机器上同样可复现。同一个 `Generator` 每次调用都会推进内部状态，连续调用不会产生重复序列。
+
 #### 5.7 FFT
 
 ```python
@@ -365,6 +371,81 @@ print(np.isinf(a))
 print(np.isfinite(a))
 ```
 
+#### 5.11 日期时间与网格
+
+```python
+import rsnumpy as np
+
+# 日期时间标量（兼容 numpy.datetime64）
+d = np.datetime64('2024-01-01')
+delta = np.timedelta64(7, 'D')          # 7 天
+print(d + delta)                         # numpy.datetime64('2024-01-08')
+print(np.datetime64('2024-01-08') - d)   # numpy.timedelta64(7,'D')
+
+# 用 datetime64/timedelta64 生成日期序列
+days = np.arange(np.datetime64('2024-01-01'),
+                 np.datetime64('2024-01-05'),
+                 np.timedelta64(1, 'D'))
+
+# 坐标网格（默认 'xy' 索引，与 NumPy 一致）
+x = np.array([1, 2, 3])
+y = np.array([10, 20])
+X, Y = np.meshgrid(x, y)
+print(X.tolist())   # [[1.0, 2.0, 3.0], [1.0, 2.0, 3.0]]
+print(Y.tolist())   # [[10.0, 10.0, 10.0], [20.0, 20.0, 20.0]]
+```
+
+#### 5.12 补充 API 覆盖（对齐 NumPy 2.5.1）
+
+在核心 Rust 原语之上，`_extra.py` 补充了约 140 个 NumPy 兼容函数，全部对照 NumPy 2.5.1 逐函数验证通过。计算密集/逐元素部分复用现有 Rust 原语，纯变形与组合类则以薄 Python 包装实现。
+
+```python
+import rsnumpy as np
+
+# 三角/双曲别名（Array API 命名）
+np.acos(1.0); np.asin(0.0); np.atan(1.0); np.atan2(1.0, 1.0)
+np.radians(180.0); np.degrees(np.pi)
+
+# 逐元素数学
+np.hypot(3.0, 4.0)          # 5.0
+np.maximum([1, 5], [3, 2])  # [3, 5]
+np.fmax([np.nan, 2], [1, np.nan])  # 忽略 NaN
+np.logaddexp(0.0, 0.0)      # 数值稳定 log(exp(a)+exp(b))
+np.rint([0.5, 1.5, 2.5])    # 四舍六入五成双 -> [0, 2, 2]
+np.gcd(12, 8); np.lcm(4, 6)
+frac, whole = np.modf([1.5, 2.25])
+
+# nan 系列归约
+a = np.array([1.0, np.nan, 3.0])
+np.nansum(a); np.nanmean(a); np.nanstd(a); np.nanmax(a)
+np.nancumsum(a); np.nanargmax(a)
+np.prod([1, 2, 3, 4])       # 24
+
+# 集合运算
+np.intersect1d([1, 2, 3], [2, 3, 4])  # [2, 3]
+np.union1d([1, 2], [2, 3])            # [1, 2, 3]
+np.setdiff1d([1, 2, 3], [2])          # [1, 3]
+np.isin([1, 2, 5], [1, 5])            # [True, False, True]
+uniq, counts = np.unique_counts([1, 1, 2, 3, 3, 3])
+
+# 变形 / 组合
+np.block([[np.eye(2), np.ones((2, 1))]])
+np.pad([1, 2, 3], (1, 2), mode='reflect')
+np.kron([1, 2], [1, 1])
+np.tensordot(np.ones((2, 3)), np.ones((3, 4)), axes=1)
+
+# 复数
+z = np.array([1 + 2j, 3 - 1j])
+np.real(z); np.imag(z); np.conjugate(z); np.angle(z)
+
+# 信号 / 窗函数 / 插值
+np.convolve([1, 2, 3], [1, 1])
+np.interp(2.5, [1, 2, 3], [10, 20, 30])
+np.hanning(8); np.hamming(8); np.blackman(8); np.bartlett(8)
+np.vander([1, 2, 3], 3)
+np.bincount([0, 1, 1, 2, 2, 2])
+```
+
 ---
 
 ### 6. 常见问题
@@ -457,26 +538,32 @@ A: 当前版本仅支持 CPU。
 **Project layout:**
 
 ```
+
 rsnumpy/
-├── src/                       # Rust source
-│   ├── lib.rs                 # Core ndarray & general functions
-│   ├── fft.rs                 # Fast Fourier Transform
-│   ├── linalg.rs              # Linear algebra
-│   └── random.rs              # Random number generation
-├── rsnumpy/                     # Python thin wrappers
-│   ├── __init__.py            # Main module, exports public API
-│   ├── array_methods.py       # ndarray object methods
-│   ├── array_ops.py           # Array manipulation functions
-│   ├── math_functions.py      # Math functions
-│   ├── statistics.py          # Statistics functions
-│   ├── io.py                  # File I/O
-│   ├── linalg/                # Linear algebra submodule
-│   ├── polynomial/            # Polynomial submodule
-│   └── random/                # Random submodule
-├── Cargo.toml                 # Rust dependencies
-├── pyproject.toml             # Python build config
-├── build_wheel.sh             # Build script
-└── README.md                  # This file
+├── src/ # Rust source
+│ ├── lib.rs # Core ndarray & general functions
+│ ├── indexing.rs # Multi-dimensional indexing & slicing
+│ ├── fft.rs # Fast Fourier Transform
+│ ├── linalg.rs # Linear algebra
+│ └── random.rs # Random number generation (reproducible parallel sampling)
+├── python/rsnumpy/ # Python thin wrappers
+│ ├── **init**.py # Main module, exports public API
+│ ├── array_methods.py # ndarray object methods
+│ ├── array_ops.py # Array manipulation functions
+│ ├── _extra.py # Supplementary API (aliases, nan-reductions, set ops, windows, ...)
+│ ├── math_functions.py # Math functions
+│ ├── statistics.py # Statistics functions
+│ ├── char.py # String array functions
+│ ├── matlib.py # Matrix construction helpers
+│ ├── io.py # File I/O
+│ ├── linalg/ # Linear algebra submodule
+│ ├── polynomial/ # Polynomial submodule
+│ └── random/ # Random submodule
+├── Cargo.toml # Rust dependencies
+├── pyproject.toml # Python build config
+├── build_wheel.sh # Build script
+└── README.md # This file
+
 ```
 
 ---
@@ -692,6 +779,8 @@ print(np.random.rand(3).tolist())
 print(np.random.randn(3).tolist())
 ```
 
+> **Reproducibility**: the same `seed` always produces the same result. Large-array sampling runs in parallel on the Rust side using fixed-size chunks with per-chunk seeds derived via `splitmix64`, so results are **independent of the thread count** and reproducible across machines. Each call on a `Generator` advances its internal state, so consecutive calls never repeat the same sequence.
+
 #### 5.7 FFT
 
 ```python
@@ -774,6 +863,81 @@ print(np.isinf(a))
 print(np.isfinite(a))
 ```
 
+#### 5.11 Datetime & meshgrid
+
+```python
+import rsnumpy as np
+
+# Datetime scalars (compatible with numpy.datetime64)
+d = np.datetime64('2024-01-01')
+delta = np.timedelta64(7, 'D')           # 7 days
+print(d + delta)                          # numpy.datetime64('2024-01-08')
+print(np.datetime64('2024-01-08') - d)    # numpy.timedelta64(7,'D')
+
+# Generate a date range with datetime64/timedelta64
+days = np.arange(np.datetime64('2024-01-01'),
+                 np.datetime64('2024-01-05'),
+                 np.timedelta64(1, 'D'))
+
+# Coordinate grids (default 'xy' indexing, matching NumPy)
+x = np.array([1, 2, 3])
+y = np.array([10, 20])
+X, Y = np.meshgrid(x, y)
+print(X.tolist())   # [[1.0, 2.0, 3.0], [1.0, 2.0, 3.0]]
+print(Y.tolist())   # [[10.0, 10.0, 10.0], [20.0, 20.0, 20.0]]
+```
+
+#### 5.12 Supplementary API Coverage (aligned with NumPy 2.5.1)
+
+On top of the core Rust primitives, `_extra.py` adds ~140 NumPy-compatible functions, each verified against NumPy 2.5.1. Compute-heavy / element-wise paths reuse existing Rust primitives, while pure reshaping and composition helpers are implemented as thin Python wrappers.
+
+```python
+import rsnumpy as np
+
+# Trig/hyperbolic aliases (Array API names)
+np.acos(1.0); np.asin(0.0); np.atan(1.0); np.atan2(1.0, 1.0)
+np.radians(180.0); np.degrees(np.pi)
+
+# Element-wise math
+np.hypot(3.0, 4.0)          # 5.0
+np.maximum([1, 5], [3, 2])  # [3, 5]
+np.fmax([np.nan, 2], [1, np.nan])  # NaN-ignoring
+np.logaddexp(0.0, 0.0)      # numerically stable log(exp(a)+exp(b))
+np.rint([0.5, 1.5, 2.5])    # round-half-to-even -> [0, 2, 2]
+np.gcd(12, 8); np.lcm(4, 6)
+frac, whole = np.modf([1.5, 2.25])
+
+# nan-reductions
+a = np.array([1.0, np.nan, 3.0])
+np.nansum(a); np.nanmean(a); np.nanstd(a); np.nanmax(a)
+np.nancumsum(a); np.nanargmax(a)
+np.prod([1, 2, 3, 4])       # 24
+
+# Set operations
+np.intersect1d([1, 2, 3], [2, 3, 4])  # [2, 3]
+np.union1d([1, 2], [2, 3])            # [1, 2, 3]
+np.setdiff1d([1, 2, 3], [2])          # [1, 3]
+np.isin([1, 2, 5], [1, 5])            # [True, False, True]
+uniq, counts = np.unique_counts([1, 1, 2, 3, 3, 3])
+
+# Reshaping / composition
+np.block([[np.eye(2), np.ones((2, 1))]])
+np.pad([1, 2, 3], (1, 2), mode='reflect')
+np.kron([1, 2], [1, 1])
+np.tensordot(np.ones((2, 3)), np.ones((3, 4)), axes=1)
+
+# Complex
+z = np.array([1 + 2j, 3 - 1j])
+np.real(z); np.imag(z); np.conjugate(z); np.angle(z)
+
+# Signal / windows / interpolation
+np.convolve([1, 2, 3], [1, 1])
+np.interp(2.5, [1, 2, 3], [10, 20, 30])
+np.hanning(8); np.hamming(8); np.blackman(8); np.bartlett(8)
+np.vander([1, 2, 3], 3)
+np.bincount([0, 1, 1, 2, 2, 2])
+```
+
 ---
 
 ### 6. FAQ