nt25 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

nt25/__init__.py

@@ -1,7 +1,7 @@
-import importlib.metadata as meta
-
-from .lib import fio, calc, draw
-from .lib.draw import DType
-
-__version__ = meta.version(str(__package__))
-__all__ = ('__version__', 'fio', 'calc', 'draw', 'DType')
+import importlib.metadata as meta
+
+from .lib import fio, calc, draw
+from .lib.draw import DType
+
+__version__ = meta.version(str(__package__))
+__all__ = ('__version__', 'fio', 'calc', 'draw', 'DType')

nt25/lib/calc.py

@@ -1,104 +1,104 @@
-import re
-from typing import Iterable
-import numpy as np
-
-from sympy import symbols, Eq, solve
-
-from sklearn.metrics import r2_score as r2
-from sklearn.linear_model import LinearRegression
-from sklearn.preprocessing import PolynomialFeatures
-
-
-def str2e(s) -> str:
-  return s if len(s) < 5 else f'{float(s):.4e}'
-
-
-def fl2el(fl) -> list[str]:
-  return list(map(lambda f: str2e(str(f)), fl))
-
-
-def poly(x, y, degree=2) -> Iterable:
-  return np.polyfit(x, y, degree).tolist()
-
-
-def polyResults(coef, x) -> Iterable:
-  return np.poly1d(coef)(x).tolist()
-
-
-def polyRoots(coef) -> Iterable:
-  return np.roots(coef).tolist()
-
-
-def xn2y(X, y, degree=2, output=False, foo='xn2y'):
-  poly = PolynomialFeatures(degree=degree, include_bias=False)
-
-  Xa = np.array(X)
-  if len(Xa.shape) == 1:
-    Xa = np.array([X])
-
-  x = Xa.T
-  xp = poly.fit_transform(x)
-
-  model = LinearRegression()
-  model.fit(xp, y)
-  # y1 = model.predict(xp)
-
-  coef = model.coef_
-  names = poly.get_feature_names_out().tolist()
-
-  eq = f'{model.intercept_:.4e} + '
-  for i in range(len(names)):
-    ex = names[i].replace('^', '**').replace(' ', '*')
-    eq += f"({coef[i]:.4e})*{ex} + "
-
-  eq = re.sub(r'x(\d+)', r'x[\1]', eq[:-3])
-  func = _genFunc(foo, 'x', eq)
-
-  y2 = [func(x[i]) for i in range(len(x))]
-  yr2 = r2(y, y2)
-
-  if output:
-    q = eq.replace(' + ', ' + \\\n\t') \
-        .replace('*x', ' * x').replace('e+00', '')
-
-    print(f'{foo}: degree = {degree}, R² = {yr2:.4%}\n  y = {q}\n')
-
-  return {
-      'func': func,
-      'r2': yr2,
-      'eq': eq,
-  }
-
-
-def _genFunc(name, args, expr):
-  local = {}
-  src = compile(f'def {name}({args}):\n\treturn {expr}', "<string>", "exec")
-  exec(src, {}, local)
-  return local[name]
-
-
-def solveEq(eq, output=False, foo='solve'):
-  (x0, x1, y) = symbols('x:2,y')
-  eq = re.sub(r'x\[(\d+)\]', r'x\1', eq)
-  solution = solve(Eq(eval(eq), y), x0)
-
-  real = []
-  for s in solution:
-    sol = str(s)
-    if 'I' in sol:
-      continue
-
-    func = _genFunc(foo, 'y, x1=0', sol)
-    s4e = re.sub(r'([-]?\d*\.\d+)', lambda m: str2e(m.group(0)), sol)
-
-    if output:
-      print(f'{foo}: {eq}\n  x0 = {s4e}\n')
-
-    real.append({
-        'org': eq,
-        'func': func,
-        'eq': sol,
-        'eq.4e': s4e,
-    })
-
-  return real
+import re
+from typing import Iterable
+import numpy as np
+
+from sympy import symbols, Eq, solve
+
+from sklearn.metrics import r2_score as r2
+from sklearn.linear_model import LinearRegression
+from sklearn.preprocessing import PolynomialFeatures
+
+
+def str2e(s) -> str:
+  return s if len(s) < 5 else f'{float(s):.4e}'
+
+
+def fl2el(fl) -> list[str]:
+  return list(map(lambda f: str2e(str(f)), fl))
+
+
+def poly(x, y, degree=2) -> Iterable:
+  return np.polyfit(x, y, degree).tolist()
+
+
+def polyResults(coef, x) -> Iterable:
+  return np.poly1d(coef)(x).tolist()
+
+
+def polyRoots(coef) -> Iterable:
+  return np.roots(coef).tolist()
+
+
+def xn2y(X, y, degree=2, output=False, foo='xn2y'):
+  poly = PolynomialFeatures(degree=degree, include_bias=False)
+
+  Xa = np.array(X)
+  if len(Xa.shape) == 1:
+    Xa = np.array([X])
+
+  x = Xa.T
+  xp = poly.fit_transform(x)
+
+  model = LinearRegression()
+  model.fit(xp, y)
+  # y1 = model.predict(xp)
+
+  coef = model.coef_
+  names = poly.get_feature_names_out().tolist()
+
+  eq = f'{model.intercept_:.4e} + '
+  for i in range(len(names)):
+    ex = names[i].replace('^', '**').replace(' ', '*')
+    eq += f"({coef[i]:.4e})*{ex} + "
+
+  eq = re.sub(r'x(\d+)', r'x[\1]', eq[:-3])
+  func = _genFunc(foo, 'x', eq)
+
+  y2 = [func(x[i]) for i in range(len(x))]
+  yr2 = r2(y, y2)
+
+  if output:
+    q = eq.replace(' + ', ' + \\\n\t') \
+        .replace('*x', ' * x').replace('e+00', '')
+
+    print(f'{foo}: degree = {degree}, R² = {yr2:.4%}\n  y = {q}\n')
+
+  return {
+      'func': func,
+      'r2': yr2,
+      'eq': eq,
+  }
+
+
+def _genFunc(name, args, expr):
+  local = {}
+  src = compile(f'def {name}({args}):\n\treturn {expr}', "<string>", "exec")
+  exec(src, {}, local)
+  return local[name]
+
+
+def solveEq(eq, output=False, foo='solve'):
+  (x0, x1, y) = symbols('x:2,y')
+  eq = re.sub(r'x\[(\d+)\]', r'x\1', eq)
+  solution = solve(Eq(eval(eq), y), x0)
+
+  real = []
+  for s in solution:
+    sol = str(s)
+    if 'I' in sol:
+      continue
+
+    func = _genFunc(foo, 'y, x1=0', sol)
+    s4e = re.sub(r'([-]?\d*\.\d+)', lambda m: str2e(m.group(0)), sol)
+
+    if output:
+      print(f'{foo}: {eq}\n  x0 = {s4e}\n')
+
+    real.append({
+        'org': eq,
+        'func': func,
+        'eq': sol,
+        'eq.4e': s4e,
+    })
+
+  return real

nt25/lib/draw.py

@@ -1,141 +1,141 @@
-from enum import Enum
-from random import sample
-
-import numpy as np
-
-from matplotlib.axes import Axes
-from matplotlib import pyplot as plot
-
-# import matplotlib.animation as animation
-
-
-FONTS = ['PingFang SC', 'Microsoft YaHei', 'Arial']
-COLORS = ['blue', 'red', 'orange', 'black', 'pink']
-
-
-def _getPlot(ref, pos, font='Roboto') -> Axes:
-  if 'figure' not in ref:
-    ref['figure'] = plot.figure()
-    plot.rcParams['font.sans-serif'] = [font] + FONTS
-    # plot.rcParams['font.sans-serif'] = ['Arial']
-    plot.rcParams['axes.unicode_minus'] = False
-
-  if 'subplot' not in ref:
-    ref['subplot'] = {}
-
-  fig = ref['figure']
-  sub = ref['subplot']
-
-  if pos not in sub:
-    sub[pos] = fig.add_subplot(pos)
-
-  return sub[pos]
-
-
-def _genList(refer: list, length, random=False):
-  r = []
-
-  while len(r) < length:
-    r += sample(refer, len(refer)) if random else refer
-
-  return r[:length]
-
-
-def _coord(ref, X, Y, color, pos, randomColor, method, *args, **kwargs):
-  Xa = np.array(X)
-  Ya = np.array(Y)
-
-  if Xa.shape != Ya.shape:
-    print(f"bad shape {Xa.shape} != {Ya.shape}")
-    return
-
-  count = 1
-  # count = Xa.shape[0] if len(Xa.shape) > 1 else 1
-
-  if len(Xa.shape) > 1:
-    count = Xa.shape[0]
-    if count == 1:
-      X = X[0]
-      Y = Y[0]
-
-  if color is None:
-    color = _genList(COLORS, count, random=randomColor)
-
-  if count == 1:
-    if isinstance(color, (list, tuple)):
-      color = color[0]
-  elif len(color) != count:
-    print(f"bad color.len {len(color)} != {count}")
-    return
-
-  if pos is None:
-    pos = [111] * count
-
-  if count == 1:
-    if isinstance(pos, (list, tuple)):
-      pos = pos[0]
-  elif len(pos) != count:
-    print(f"bad pos.len {len(pos)} != {count}")
-    return
-
-  if count > 1 and isinstance(pos, list):
-    for i in range(count):
-      p = _getPlot(ref, pos[i])
-      method(p, X[i], Y[i], color=color[i], *args, **kwargs)
-  else:
-    p = _getPlot(ref, pos)
-    method(p, X, Y, color=color, *args, **kwargs)
-
-  return ref
-
-
-class DType(Enum):
-  scatter = 1,
-  line = 2,
-  func = 3,
-
-
-def d2d(type=DType.scatter, X=None, Y=None, Func=None, min=None, max=None,
-        ref=None, color=None, pos=None, randomColor=False, show=False,
-        *args, **kwargs):
-  if ref is None:
-    ref = {}
-
-  match type:
-    case DType.scatter:
-      func = Axes.scatter
-
-    case DType.line:
-      func = Axes.plot
-
-    case DType.func:
-      func = Axes.plot
-      if Func is not None and min is not None and max is not None:
-        if callable(Func):
-          Func = (Func,)
-
-        X = []
-        Y = []
-
-        for i in range(len(Func)):
-          dx = np.linspace(min[i] if isinstance(min, (list, tuple)) else min,
-                           max[i] if isinstance(max, (list, tuple)) else max)
-
-          X.append(dx)
-          Y.append([Func[i]([x]) for x in dx])
-
-  ref = _coord(ref, X, Y, color=color, pos=pos, randomColor=randomColor,
-               method=func, *args, **kwargs)
-
-  if show:
-    plot.show()
-
-  return ref
-
-
-def show():
-  plot.show()
-
-
-def clear():
-  plot.clf()
+from enum import Enum
+from random import sample
+
+import numpy as np
+
+from matplotlib.axes import Axes
+from matplotlib import pyplot as plot
+
+# import matplotlib.animation as animation
+
+
+FONTS = ['PingFang SC', 'Microsoft YaHei', 'Arial']
+COLORS = ['blue', 'red', 'orange', 'black', 'pink']
+
+
+def _getPlot(ref, pos, font='Roboto') -> Axes:
+  if 'figure' not in ref:
+    ref['figure'] = plot.figure()
+    plot.rcParams['font.sans-serif'] = [font] + FONTS
+    # plot.rcParams['font.sans-serif'] = ['Arial']
+    plot.rcParams['axes.unicode_minus'] = False
+
+  if 'subplot' not in ref:
+    ref['subplot'] = {}
+
+  fig = ref['figure']
+  sub = ref['subplot']
+
+  if pos not in sub:
+    sub[pos] = fig.add_subplot(pos)
+
+  return sub[pos]
+
+
+def _genList(refer: list, length, random=False):
+  r = []
+
+  while len(r) < length:
+    r += sample(refer, len(refer)) if random else refer
+
+  return r[:length]
+
+
+def _coord(ref, X, Y, color, pos, randomColor, method, *args, **kwargs):
+  Xa = np.array(X)
+  Ya = np.array(Y)
+
+  if Xa.shape != Ya.shape:
+    print(f"bad shape {Xa.shape} != {Ya.shape}")
+    return
+
+  count = 1
+  # count = Xa.shape[0] if len(Xa.shape) > 1 else 1
+
+  if len(Xa.shape) > 1:
+    count = Xa.shape[0]
+    if count == 1:
+      X = X[0]
+      Y = Y[0]
+
+  if color is None:
+    color = _genList(COLORS, count, random=randomColor)
+
+  if count == 1:
+    if isinstance(color, (list, tuple)):
+      color = color[0]
+  elif len(color) != count:
+    print(f"bad color.len {len(color)} != {count}")
+    return
+
+  if pos is None:
+    pos = [111] * count
+
+  if count == 1:
+    if isinstance(pos, (list, tuple)):
+      pos = pos[0]
+  elif len(pos) != count:
+    print(f"bad pos.len {len(pos)} != {count}")
+    return
+
+  if count > 1 and isinstance(pos, list):
+    for i in range(count):
+      p = _getPlot(ref, pos[i])
+      method(p, X[i], Y[i], color=color[i], *args, **kwargs)
+  else:
+    p = _getPlot(ref, pos)
+    method(p, X, Y, color=color, *args, **kwargs)
+
+  return ref
+
+
+class DType(Enum):
+  scatter = 1,
+  line = 2,
+  func = 3,
+
+
+def d2d(type=DType.scatter, X=None, Y=None, Func=None, min=None, max=None,
+        ref=None, color=None, pos=None, randomColor=False, show=False,
+        *args, **kwargs):
+  if ref is None:
+    ref = {}
+
+  match type:
+    case DType.scatter:
+      func = Axes.scatter
+
+    case DType.line:
+      func = Axes.plot
+
+    case DType.func:
+      func = Axes.plot
+      if Func is not None and min is not None and max is not None:
+        if callable(Func):
+          Func = (Func,)
+
+        X = []
+        Y = []
+
+        for i in range(len(Func)):
+          dx = np.linspace(min[i] if isinstance(min, (list, tuple)) else min,
+                           max[i] if isinstance(max, (list, tuple)) else max)
+
+          X.append(dx)
+          Y.append([Func[i]([x]) for x in dx])
+
+  ref = _coord(ref, X, Y, color=color, pos=pos, randomColor=randomColor,
+               method=func, *args, **kwargs)
+
+  if show:
+    plot.show()
+
+  return ref
+
+
+def show():
+  plot.show()
+
+
+def clear():
+  plot.clf()

nt25/lib/fio.py

@@ -1,116 +1,116 @@
-import os
-import csv
-from typing import Iterable
-
-# import openpyxl
-import pandas as pd
-
-ENCODINGS = ['utf-8', 'utf-8-sig', 'gbk']
-
-
-def _switchEnc(foo, encoding=None, *args, **kwargs):
-  result = None
-
-  if encoding is None:
-    for enc in ENCODINGS if os.name != 'nt' else reversed(ENCODINGS):
-      try:
-        result = foo(encoding=enc, *args, **kwargs)
-        break
-      except Exception as e:
-        print(e)
-
-  else:
-    result = foo(encoding=encoding, *args, **kwargs)
-
-  return result
-
-
-def _getCSV(file, width, startLine, startCol, encoding):
-  f = open(file, encoding=encoding)
-  cf = csv.reader(f)
-
-  count = 0
-  result = [[] for _ in range(width - startCol)]
-
-  for line in cf:
-    if count >= startLine:
-      maxWidth = len(line)
-      for i in range(startCol, width):
-        x = line[i].strip() if maxWidth > i else ''
-
-        try:
-          result[i - startCol].append(float(x))
-        except ValueError:
-          result[i - startCol].append(x)
-
-    count += 1
-
-  f.close()
-  return result
-
-
-def _getCSV2(file, encoding=None, colsInline=True):
-  df = pd.read_csv(file, encoding=encoding)
-
-  row = df.to_numpy()
-  r = row.T if colsInline else row
-
-  return r.tolist()
-
-
-def getCSV(file, width=2, startLine=1, startCol=0, encoding=None):
-  return _switchEnc(foo=_getCSV, encoding=encoding, file=file,
-                    width=width, startLine=startLine, startCol=startCol)
-
-
-def getCSV2(file, encoding=None):
-  return _switchEnc(foo=_getCSV2, encoding=encoding, file=file)
-
-
-def saveCSV(data, file, encoding=None, colsInline=True):
-  if encoding is None:
-    encoding = 'utf-8'
-
-  with open(file, 'w', newline='', encoding=encoding) as f:
-    content = []
-
-    for d in data:
-      if isinstance(d, str):
-        content.append(data[d])
-      elif isinstance(d, Iterable):
-        content.append(d)
-
-    if colsInline:
-      content = list(map(list, zip(*content)))
-
-    cf = csv.writer(f)
-    cf.writerows(content)
-
-
-def saveCSV2(data, file, encoding=None, float_format=None, colsInline=True):
-  if data is not None:
-    df = pd.DataFrame(data)
-
-    if colsInline:
-      df = df.T
-
-    df.to_csv(file, index=False, encoding=encoding, float_format=float_format)
-
-
-def getXlsx(file, sheet=0, colsInLine=True):
-  df = pd.read_excel(file, sheet_name=sheet)
-
-  row = df.to_numpy()
-  r = row.T if colsInLine else row
-
-  return r.tolist()
-
-
-def saveXlsx(data, file, colsInline=True):
-  if data is not None:
-    df = pd.DataFrame(data)
-
-    if colsInline:
-      df = df.T
-
-    df.to_excel(file, index=False)
+import os
+import csv
+from typing import Iterable
+
+# import openpyxl
+import pandas as pd
+
+ENCODINGS = ['utf-8', 'utf-8-sig', 'gbk']
+
+
+def _switchEnc(foo, encoding=None, *args, **kwargs):
+  result = None
+
+  if encoding is None:
+    for enc in ENCODINGS if os.name != 'nt' else reversed(ENCODINGS):
+      try:
+        result = foo(encoding=enc, *args, **kwargs)
+        break
+      except Exception as e:
+        print(e)
+
+  else:
+    result = foo(encoding=encoding, *args, **kwargs)
+
+  return result
+
+
+def _getCSV(file, width, startLine, startCol, encoding):
+  f = open(file, encoding=encoding)
+  cf = csv.reader(f)
+
+  count = 0
+  result = [[] for _ in range(width - startCol)]
+
+  for line in cf:
+    if count >= startLine:
+      maxWidth = len(line)
+      for i in range(startCol, width):
+        x = line[i].strip() if maxWidth > i else ''
+
+        try:
+          result[i - startCol].append(float(x))
+        except ValueError:
+          result[i - startCol].append(x)
+
+    count += 1
+
+  f.close()
+  return result
+
+
+def _getCSV2(file, encoding=None, colsInline=True):
+  df = pd.read_csv(file, encoding=encoding)
+
+  row = df.to_numpy()
+  r = row.T if colsInline else row
+
+  return r.tolist()
+
+
+def getCSV(file, width=2, startLine=1, startCol=0, encoding=None):
+  return _switchEnc(foo=_getCSV, encoding=encoding, file=file,
+                    width=width, startLine=startLine, startCol=startCol)
+
+
+def getCSV2(file, encoding=None):
+  return _switchEnc(foo=_getCSV2, encoding=encoding, file=file)
+
+
+def saveCSV(data, file, encoding=None, colsInline=True):
+  if encoding is None:
+    encoding = 'utf-8'
+
+  with open(file, 'w', newline='', encoding=encoding) as f:
+    content = []
+
+    for d in data:
+      if isinstance(d, str):
+        content.append(data[d])
+      elif isinstance(d, Iterable):
+        content.append(d)
+
+    if colsInline:
+      content = list(map(list, zip(*content)))
+
+    cf = csv.writer(f)
+    cf.writerows(content)
+
+
+def saveCSV2(data, file, encoding=None, float_format=None, colsInline=True):
+  if data is not None:
+    df = pd.DataFrame(data)
+
+    if colsInline:
+      df = df.T
+
+    df.to_csv(file, index=False, encoding=encoding, float_format=float_format)
+
+
+def getXlsx(file, sheet=0, colsInLine=True):
+  df = pd.read_excel(file, sheet_name=sheet)
+
+  row = df.to_numpy()
+  r = row.T if colsInLine else row
+
+  return r.tolist()
+
+
+def saveXlsx(data, file, colsInline=True):
+  if data is not None:
+    df = pd.DataFrame(data)
+
+    if colsInline:
+      df = df.T
+
+    df.to_excel(file, index=False)

nt25/main.py

@@ -1,4 +1,4 @@
-from ntpy import fio, calc, draw, DType, __version__
+from nt25 import fio, calc, draw, DType, __version__
 
 # import timeit
 # timeit.timeit('', number=100, globals=globals())

{nt25-0.1.0.dist-info → nt25-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nt25
-Version: 0.1.0
+Version: 0.1.1
 Summary: Neo's Tools of Python
 Requires-Python: >=3.13
 Requires-Dist: matplotlib>=3.10.6
@@ -10,14 +10,15 @@ Requires-Dist: scikit-learn>=1.7.1
 Requires-Dist: sympy>=1.14.0
 Description-Content-Type: text/markdown
 
-# NT.py
+# nt25.py
 
-Neo's Tools of Python
+Neo's Tools of Python in 2025
 
 ## plans
 
 - [x] fio
 - [x] calc
+- [x] draw with 2d
 - [ ] draw with 3d
 
 ## scripts from init

nt25-0.1.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+nt25/__init__.py,sha256=3pJhB9zM_qRUTiAj98xwwd-8muwSgGC0-Sd_CEYi4Tc,200
+nt25/main.py,sha256=TkgLVeEm3GENii-JVUhLVIfO31N9QLjwkRbELxj1k4A,1256
+nt25/lib/calc.py,sha256=FHPHtHw7Mcm3PlVoC1c3ghpapZe0GhPVG5-mmfHkDaw,2242
+nt25/lib/draw.py,sha256=5llf6Q3ljBNBkIEphZzL7Xv0ZZu8i2CTBHYmRzH84lU,3020
+nt25/lib/fio.py,sha256=WvHpG6QYR1NE19Ss3Sy2FdajTxibX5SVW3PyC5Y5Krk,2525
+nt25-0.1.1.dist-info/METADATA,sha256=b9YUoAt5tU5DL5DfstJUFV175yQLrCLjn-o6hbya_gs,506
+nt25-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nt25-0.1.1.dist-info/entry_points.txt,sha256=mApCFA2DAwrQrcWa5JqcKq7lPzogm_Yi4EejBUo-Of8,40
+nt25-0.1.1.dist-info/RECORD,,

nt25-0.1.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+nt25 = nt25.main:main

nt25-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-nt25/__init__.py,sha256=VB6BTxQeAy3eAqEzpe2x0Kc4w-TGakvtZPYzBPq8ldY,207
-nt25/main.py,sha256=Zf0bOV3YciAb8kTbljR3C9wEspUJ4jcexjTEXeiP5vU,1256
-nt25/lib/calc.py,sha256=HHjp9bIEszLloB1TurnMSNqk0yo6Jyd7egMEeoQIK3o,2346
-nt25/lib/draw.py,sha256=HY6O1r-NQmue24gG5WHTocq4gDPpM-qconoy-Ozfs7g,3161
-nt25/lib/fio.py,sha256=ujt4AbNUX6RyjHwqUn2AXly9ZWjgslzhqdY2DK7Q4Lo,2641
-nt25-0.1.0.dist-info/METADATA,sha256=oGVEI_Ewpob8YuIfvyMM9uUE3PDRXZaQl9yvXRVKSBI,477
-nt25-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nt25-0.1.0.dist-info/entry_points.txt,sha256=jymrdQKvCJUGsJFD51GtyJ0AEBAzpiXagA6NVwYuUmw,40
-nt25-0.1.0.dist-info/RECORD,,

nt25-0.1.0.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-ntpy = ntpy.main:main