easyfuncplot 0.1.1__tar.gz

easyfuncplot-0.1.1/License

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 aicodebearcm1
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

easyfuncplot-0.1.1/PKG-INFO

@@ -0,0 +1,30 @@
+Metadata-Version: 2.4
+Name: easyfuncplot
+Version: 0.1.1
+Summary: Plot mathematical functions from strings or Python callables
+Author-email: Your Name <you@example.com>
+License: MIT
+Project-URL: Homepage, https://github.com/yourname/mathplot
+Keywords: math,plot,function,matplotlib
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: License
+Requires-Dist: matplotlib>=3.5
+Requires-Dist: numpy>=1.21
+Provides-Extra: dev
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: build; extra == "dev"
+Requires-Dist: twine; extra == "dev"
+Dynamic: license-file
+
+# mathplot
+
+Plot mathematical functions from strings or Python callables.
+
+## 安装
+
+```bash
+pip install easyfuncplot

easyfuncplot-0.1.1/README.md

@@ -0,0 +1,8 @@
+# mathplot
+
+Plot mathematical functions from strings or Python callables.
+
+## 安装
+
+```bash
+pip install easyfuncplot

easyfuncplot-0.1.1/pyproject.toml

@@ -0,0 +1,31 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "easyfuncplot"
+version = "0.1.1"
+description = "Plot mathematical functions from strings or Python callables"
+readme = "README.md"
+requires-python = ">=3.9"
+license = { text = "MIT" }
+authors = [{ name = "Your Name", email = "you@example.com" }]
+keywords = ["math", "plot", "function", "matplotlib"]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+dependencies = [
+    "matplotlib>=3.5",
+    "numpy>=1.21",
+]
+
+[project.optional-dependencies]
+dev = ["pytest", "build", "twine"]
+
+[project.urls]
+Homepage = "https://github.com/yourname/mathplot"
+
+[tool.setuptools.packages.find]
+where = ["src"]

easyfuncplot-0.1.1/setup.cfg

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

easyfuncplot-0.1.1/src/easyfuncplot.egg-info/PKG-INFO

@@ -0,0 +1,30 @@
+Metadata-Version: 2.4
+Name: easyfuncplot
+Version: 0.1.1
+Summary: Plot mathematical functions from strings or Python callables
+Author-email: Your Name <you@example.com>
+License: MIT
+Project-URL: Homepage, https://github.com/yourname/mathplot
+Keywords: math,plot,function,matplotlib
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: License
+Requires-Dist: matplotlib>=3.5
+Requires-Dist: numpy>=1.21
+Provides-Extra: dev
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: build; extra == "dev"
+Requires-Dist: twine; extra == "dev"
+Dynamic: license-file
+
+# mathplot
+
+Plot mathematical functions from strings or Python callables.
+
+## 安装
+
+```bash
+pip install easyfuncplot

easyfuncplot-0.1.1/src/easyfuncplot.egg-info/SOURCES.txt

@@ -0,0 +1,14 @@
+License
+README.md
+pyproject.toml
+src/easyfuncplot.egg-info/PKG-INFO
+src/easyfuncplot.egg-info/SOURCES.txt
+src/easyfuncplot.egg-info/dependency_links.txt
+src/easyfuncplot.egg-info/requires.txt
+src/easyfuncplot.egg-info/top_level.txt
+src/mathplot/_init_.py
+src/mathplot/auto_range.py
+src/mathplot/parser.py
+src/mathplot/plotter.py
+src/mathplot/sampler.py
+tests/test_basic.py

easyfuncplot-0.1.1/src/easyfuncplot.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

easyfuncplot-0.1.1/src/easyfuncplot.egg-info/requires.txt

@@ -0,0 +1,7 @@
+matplotlib>=3.5
+numpy>=1.21
+
+[dev]
+pytest
+build
+twine

easyfuncplot-0.1.1/src/easyfuncplot.egg-info/top_level.txt

@@ -0,0 +1 @@
+mathplot

easyfuncplot-0.1.1/src/mathplot/_init_.py

@@ -0,0 +1,5 @@
+from .parser import parse
+from .plotter import plot
+
+__version__ = "0.1.0"
+__all__ = ["parse", "plot"]

easyfuncplot-0.1.1/src/mathplot/auto_range.py

@@ -0,0 +1,18 @@
+"""自动 Y 轴范围：用分位数避免单个尖峰把图压扁。"""
+import numpy as np
+from typing import Tuple
+
+
+def auto_ylim(y: np.ndarray, percentile: float = 99.5) -> Tuple[float, float]:
+    finite = y[np.isfinite(y)]
+    if finite.size == 0:
+        return -1.0, 1.0
+
+    lo = float(np.percentile(finite, 100 - percentile))
+    hi = float(np.percentile(finite, percentile))
+
+    if hi <= lo:
+        hi, lo = lo + 1.0, lo - 1.0
+
+    pad = (hi - lo) * 0.1
+    return lo - pad, hi + pad

easyfuncplot-0.1.1/src/mathplot/parser.py

@@ -0,0 +1,91 @@
+"""把字符串表达式解析成单变量函数 f(x)，不使用 eval，安全可控。"""
+import ast
+import math
+import operator
+from typing import Callable
+
+_ALLOWED_FUNCS = {
+    "sin": math.sin, "cos": math.cos, "tan": math.tan,
+    "asin": math.asin, "acos": math.acos, "atan": math.atan,
+    "sinh": math.sinh, "cosh": math.cosh, "tanh": math.tanh,
+    "exp": math.exp, "log": math.log, "log10": math.log10,
+    "sqrt": math.sqrt, "abs": abs,
+    "floor": math.floor, "ceil": math.ceil,
+    "sign": lambda v: (v > 0) - (v < 0),
+}
+
+_ALLOWED_CONSTS = {"pi": math.pi, "e": math.e, "tau": math.tau}
+
+_BIN_OPS = {
+    ast.Add: operator.add,
+    ast.Sub: operator.sub,
+    ast.Mult: operator.mul,
+    ast.Div: operator.truediv,
+    ast.Pow: operator.pow,
+    ast.Mod: operator.mod,
+    ast.FloorDiv: operator.floordiv,
+}
+
+_UNARY_OPS = {ast.UAdd: operator.pos, ast.USub: operator.neg}
+
+
+def _preprocess(expr: str) -> str:
+    """把用户习惯的 ^ 转成 Python 的 **。"""
+    return expr.replace("^", "**")
+
+
+def parse(expr: str) -> Callable[[float], float]:
+    """把字符串表达式编译成单变量函数 f(x)。
+
+    >>> f = parse("x**2 + 1")
+    >>> f(3)
+    10
+    """
+    if not isinstance(expr, str):
+        raise TypeError("expr 必须是字符串")
+
+    tree = ast.parse(_preprocess(expr), mode="eval")
+
+    def _eval(node, x):
+        if isinstance(node, ast.Expression):
+            return _eval(node.body, x)
+
+        if isinstance(node, ast.Constant):
+            if isinstance(node.value, (int, float)):
+                return node.value
+            raise ValueError(f"不支持的常量: {node.value!r}")
+
+        if isinstance(node, ast.Name):
+            if node.id == "x":
+                return x
+            if node.id in _ALLOWED_CONSTS:
+                return _ALLOWED_CONSTS[node.id]
+            raise ValueError(f"未知变量: {node.id}")
+
+        if isinstance(node, ast.BinOp):
+            op = _BIN_OPS.get(type(node.op))
+            if op is None:
+                raise ValueError(f"不支持的运算符: {type(node.op).__name__}")
+            return op(_eval(node.left, x), _eval(node.right, x))
+
+        if isinstance(node, ast.UnaryOp):
+            op = _UNARY_OPS.get(type(node.op))
+            if op is None:
+                raise ValueError(f"不支持的一元运算符: {type(node.op).__name__}")
+            return op(_eval(node.operand, x))
+
+        if isinstance(node, ast.Call):
+            if not isinstance(node.func, ast.Name):
+                raise ValueError("只支持直接调用函数名")
+            fn = _ALLOWED_FUNCS.get(node.func.id)
+            if fn is None:
+                raise ValueError(f"未知函数: {node.func.id}")
+            args = [_eval(a, x) for a in node.args]
+            return fn(*args)
+
+        raise ValueError(f"不支持的语法: {type(node).__name__}")
+
+    def f(x: float) -> float:
+        return _eval(tree, x)
+
+    return f

easyfuncplot-0.1.1/src/mathplot/plotter.py

@@ -0,0 +1,83 @@
+"""渲染：把函数画成图。"""
+from typing import Callable, Optional, Union
+
+import matplotlib.pyplot as plt
+import numpy as np
+
+from .auto_range import auto_ylim
+from .parser import parse
+from .sampler import break_asymptotes, sample
+
+FuncLike = Union[str, Callable[[float], float]]
+
+
+def _to_callable(f: FuncLike) -> Callable[[float], float]:
+    if isinstance(f, str):
+        return parse(f)
+    if callable(f):
+        return f
+    raise TypeError("函数必须是字符串表达式或可调用对象")
+
+
+def _label_of(f: FuncLike) -> Optional[str]:
+    if isinstance(f, str):
+        return f
+    return getattr(f, "__name__", None)
+
+
+def plot(
+    *funcs: FuncLike,
+    xmin: float = -10,
+    xmax: float = 10,
+    n: int = 2000,
+    ylim: Optional[tuple] = None,
+    title: Optional[str] = None,
+    grid: bool = True,
+    show: bool = True,
+    save: Optional[str] = None,
+):
+    """画一个或多个函数。
+
+    例：
+        plot("x**2")
+        plot("sin(x)", "cos(x)", xmin=-6.28, xmax=6.28)
+        plot(lambda x: x**3 - 3*x)
+    """
+    if not funcs:
+        raise ValueError("至少传入一个函数")
+
+    fig, ax = plt.subplots(figsize=(8, 5))
+    all_y = []
+
+    for f in funcs:
+        fn = _to_callable(f)
+        x, y = sample(fn, xmin, xmax, n)
+        y = break_asymptotes(y)
+        ax.plot(x, y, linewidth=1.6, label=_label_of(f))
+        all_y.append(y)
+
+    if ylim is not None:
+        ax.set_ylim(*ylim)
+    elif all_y:
+        ax.set_ylim(*auto_ylim(np.concatenate(all_y)))
+
+    ax.set_xlim(xmin, xmax)
+    ax.axhline(0, color="black", linewidth=0.6)
+    ax.axvline(0, color="black", linewidth=0.6)
+    ax.grid(grid, alpha=0.3)
+    ax.set_xlabel("x")
+    ax.set_ylabel("y")
+
+    if title:
+        ax.set_title(title)
+
+    # 只在有字符串表达式（或显式命名）时显示图例
+    if any(_label_of(f) for f in funcs):
+        ax.legend()
+
+    if save:
+        fig.savefig(save, dpi=150, bbox_inches="tight")
+    if show:
+        plt.show()
+
+    return fig, ax

easyfuncplot-0.1.1/src/mathplot/sampler.py

@@ -0,0 +1,40 @@
+"""采样：把连续函数离散化，处理定义域和奇点。"""
+import numpy as np
+from typing import Callable, Tuple
+
+
+def sample(
+    f: Callable[[float], float],
+    xmin: float,
+    xmax: float,
+    n: int = 2000,
+) -> Tuple[np.ndarray, np.ndarray]:
+    """在 [xmin, xmax] 上采样 n 个点，无效点返回 nan。"""
+    if xmax <= xmin:
+        raise ValueError("xmax 必须大于 xmin")
+    if n < 2:
+        raise ValueError("n 至少为 2")
+
+    x = np.linspace(xmin, xmax, n)
+    y = np.empty_like(x)
+
+    for i, xi in enumerate(x):
+        try:
+            y[i] = float(f(float(xi)))
+        except (ValueError, ZeroDivisionError, OverflowError):
+            y[i] = np.nan
+
+    # inf / -inf 也变成 nan，让 matplotlib 自动断开
+    y = np.where(np.isfinite(y), y, np.nan)
+    return x, y
+
+
+def break_asymptotes(y: np.ndarray, threshold: float = 1e6) -> np.ndarray:
+    """在相邻两点数值突变处插入 nan，避免 1/x 之类的假竖线。"""
+    y = y.copy()
+    for i in range(1, len(y)):
+        if np.isnan(y[i - 1]) or np.isnan(y[i]):
+            continue
+        if abs(y[i] - y[i - 1]) > threshold:
+            y[i - 1] = np.nan
+    return y

easyfuncplot-0.1.1/tests/test_basic.py

@@ -0,0 +1,49 @@
+import math
+
+import pytest
+
+from mathplot import parse, plot
+from mathplot.parser import parse as parse_only
+
+
+def test_parse_basic():
+    f = parse_only("x**2 + 1")
+    assert f(3) == 10
+
+
+def test_parse_caret():
+    f = parse_only("x^2")
+    assert f(3) == 9
+
+
+def test_parse_constants():
+    f = parse_only("sin(pi/2)")
+    assert abs(f(0) - 1.0) < 1e-12
+
+
+def test_parse_rejects_unknown():
+    with pytest.raises(ValueError):
+        parse_only("__import__('os')")
+
+
+def test_parse_rejects_unknown_var():
+    with pytest.raises(ValueError):
+        parse_only("y + 1")
+
+
+def test_sample_handles_division_by_zero():
+    from mathplot.sampler import sample
+    f = parse_only("1/x")
+    x, y = sample(f, -1, 1, n=101)
+    # x=0 处应该是 nan
+    assert any(math.isnan(v) for v in y)
+
+
+def test_plot_returns_fig(monkeypatch):
+    import matplotlib
+    matplotlib.use("Agg")
+    fig, ax = plot("x**2", show=False)
+    assert fig is not None
+    assert ax is not None
+    import matplotlib.pyplot as plt
+    plt.close(fig)