xlin 0.1.35__py2.py3-none-any.whl → 0.1.37__py2.py3-none-any.whl

xlin/statistic.py

@@ -1,3 +1,4 @@
+import sys
 from typing import List, Optional
 from collections import defaultdict
 
@@ -112,8 +113,115 @@ Kurtosis: {float((data - mean).mean()**4 / std**4):.4f}\
         plt.savefig(fig_save_path, dpi=300)
     plt.show()
 
+def draw_radar(categories: list[str],
+               data: dict[str, list[float]],
+               ranges=(0, 5),
+               title: str = "Radar Chart",
+               size: tuple[float, float] = (6, 6),
+               colors: list[str] = None,
+               annotation_offset: float = 0.03,
+               annotation_kwargs: dict = None,
+               chinese_font: bool = False,
+               font_name: str = 'Heiti TC',
+               font_path: str = None,
+               fig_save_path: str = None):
+    import numpy as np
+    import matplotlib.pyplot as plt
+    from matplotlib.font_manager import FontProperties
+    """
+    Draw a radar chart with optional Chinese font support and configurable annotations.
+
+    Parameters:
+    - categories: list of dimension names.
+    - data: dict mapping series name -> list of values (same length as categories).
+    - ranges: tuple (min, max) for radial axis.
+    - title: chart title.
+    - size: figure size tuple.
+    - colors: list of color hex strings.
+    - annotation_offset: fraction of (max-min) to offset text.
+    - annotation_kwargs: additional kwargs for plt.text (font size, bbox, etc).
+    - chinese_font: whether to enable Chinese font support.
+    - font_name: name of a system font to use (overrides defaults).
+    - font_path: path to a .ttf font file (used if provided).
+    - fig_save_path: if set, saves figure to this path (PNG at 300dpi).
+    """
+    # Validate data
+    N = len(categories)
+    for label, values in data.items():
+        if len(values) != N:
+            raise ValueError(f"Values for '{label}' must have length {N}")
+
+    # Font configuration
+    if font_path:
+        prop = FontProperties(fname=font_path)
+        plt.rcParams['font.family'] = prop.get_name()
+    elif font_name:
+        plt.rcParams['font.family'] = font_name
+    elif chinese_font:
+        if sys.platform == "darwin":
+            plt.rcParams['font.family'] = ['Heiti TC', 'Arial Unicode MS']
+        elif sys.platform.startswith("win"):
+            plt.rcParams['font.family'] = ['SimHei', 'Microsoft YaHei']
+        else:
+            # Fallback to Arial Unicode
+            plt.rcParams['font.family'] = ['Arial Unicode MS']
+    else:
+        plt.rcParams['font.family'] = ['Arial']
+
+    plt.rcParams['axes.unicode_minus'] = False
+
+    # Prepare angles
+    angles = np.linspace(0, 2 * np.pi, N, endpoint=False).tolist()
+    angles += angles[:1]
+
+    # Colors
+    default_colors = ["#66c2a5", "#fc8d62", "#8da0cb", "#e78ac3", "#a6d854", "#ffd92f"]
+    colors = colors or default_colors
+
+    max_value = max(max(values) for values in data.values())
+    min_value = min(min(values) for values in data.values())
+    if ranges[0] > min_value or ranges[1] < max_value:
+        ranges = (min_value, max_value + 1)
+
+    # Create plot
+    fig, ax = plt.subplots(figsize=size, subplot_kw=dict(polar=True))
+    ax.set_theta_offset(np.pi / 2)
+    ax.set_theta_direction(-1)
+    ax.set_xticks(angles[:-1])
+    ax.set_xticklabels(categories, fontsize=12)
+    ax.set_ylim(ranges)
+    ax.set_rlabel_position(180 / N)
+    ax.grid(color='gray', linestyle='--', linewidth=0.5)
+
+    # Plot each series
+    for idx, (label, values) in enumerate(data.items()):
+        vals = values + values[:1]
+        color = colors[idx % len(colors)]
+        ax.plot(angles, vals, color=color, linewidth=2, label=label)
+        ax.fill(angles, vals, color=color, alpha=0.25)
+
+        # Annotate points
+        for angle, val in zip(angles, vals):
+            offset = (ranges[1] - ranges[0]) * annotation_offset
+            ha = 'left' if np.cos(angle) >= 0 else 'right'
+            va = 'bottom' if np.sin(angle) >= 0 else 'top'
+            txt_kwargs = dict(fontsize=10, ha=ha, va=va, bbox=dict(boxstyle="round,pad=0.3", edgecolor=color, facecolor='white', alpha=0.8))
+            if annotation_kwargs:
+                txt_kwargs.update(annotation_kwargs)
+            ax.text(angle, val + offset, f"{val}", **txt_kwargs)
+
+    ax.set_title(title, va='bottom', fontsize=16)
+    ax.legend(loc='upper right', bbox_to_anchor=(1.15, 1.05))
+    plt.tight_layout()
 
-def draw_preds_labels(preds: list[str], labels: list[str], title="Pred and Label Class Distribution", fig_save_path: Optional[str]=None):
+    if fig_save_path:
+        plt.savefig(fig_save_path, dpi=300, bbox_inches='tight')
+    plt.show()
+
+
+def draw_preds_labels(preds: list[str], labels: list[str],
+                      title="Pred and Label Class Distribution",
+                      fig_save_path: Optional[str]=None):
     from collections import Counter
     import matplotlib.pyplot as plt
 
@@ -247,6 +355,7 @@ def generate_classification_report(predictions: List[str], labels: List[str]) ->
     tail = pd.DataFrame([p], index=["precision"], columns=confusion_matrix.columns)
     confusion_matrix = pd.concat([confusion_matrix, tail], axis=0)
     confusion_matrix.index.name = "True \\ Pred"
+    confusion_matrix["sum"] = class_df["support"].values.tolist() + [class_df["support"].sum()]
     report["confusion_matrix"] = confusion_matrix
 
     micro_precision = micro_tp / (micro_tp + micro_fp) if (micro_tp + micro_fp) > 0 else 0
@@ -329,8 +438,16 @@ if __name__ == "__main__":
     truth = ["cat", "cat", "dog", "dog", "dog", "dog"]
 
     print_classification_report(preds, truth)
+    draw_preds_labels(preds, truth, title="Pred and Label Class Distribution", fig_save_path="assets/pred_label_distribution.png")
 
     import random
 
     lengths = [random.randint(0, 100) for _ in range(100)]
-    draw_histogram(lengths, bins=50, title="Length Distribution", fig_save_path="length_distribution.png")
+    draw_histogram(lengths, bins=50, title="Length Distribution", fig_save_path="assets/length_distribution.png")
+
+    categories = ['速度', '可靠性', '舒适度', '安全性', '效率', '风格', '性价比']
+    data = {
+        '产品 A': [4, 3, 2, 5, 4, 3, -1],
+        '产品 B': [3, 4, 5, 3, 3, 4, 8]
+    }
+    draw_radar(categories, data, title="产品对比雷达图", chinese_font=True, fig_save_path="assets/radar_chart.png")

{xlin-0.1.35.dist-info → xlin-0.1.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xlin
-Version: 0.1.35
+Version: 0.1.37
 Summary: toolbox for LinXueyuan
 License: MIT
 Author: LinXueyuanStdio

{xlin-0.1.35.dist-info → xlin-0.1.37.dist-info}/RECORD

@@ -4,12 +4,12 @@ xlin/jsonl.py,sha256=IDRydHh2x-8iAGCxt9ScK2wfNLNA40PxNxR5hhr4v6k,7903
 xlin/metric.py,sha256=N7wJ35y-C-IaBr1I1CJ_37lTG7gA69zmn9Xg6xSwKoI,1690
 xlin/multiprocess_mapping.py,sha256=q4EVU8JPLcRAbNf9NUEzmn8rDLDfIQ3jaW0yxVKPECk,16669
 xlin/read_as_dataframe.py,sha256=MqY57L7Wp9UoWTRlZLSBKQNaZa-dKw51-ufrKvHKf8s,9041
-xlin/statistic.py,sha256=1v5nG6aRsiFKoCYuYXxivdwJuVPLrICKDA6liFPffl0,11718
+xlin/statistic.py,sha256=nwFSN8BWfTQRimI-zfp6RwfA-I9aFDbemtV2cyh6Hq8,16533
 xlin/timing.py,sha256=XMT8dMcMolOMohDvAZOIM_BAiPMREhGQKnO1kc5s6PU,998
 xlin/util.py,sha256=TTWJaqF5D_r-gAZ_fj0kyHomvCagjwHXQZ2OPSgwd54,10976
 xlin/xls2xlsx.py,sha256=uSmXcDvIhi5Sq0LGidMXy0wErNBXdjaoa6EftYVjTXs,947
 xlin/yaml.py,sha256=kICi7G3Td5q2MaSXXt85qNTWoHMgjzt7pvn7r3C4dME,183
-xlin-0.1.35.dist-info/LICENSE,sha256=60ys6rRtc1dZOP8UjSUr9fAqhZudT3WpKe5WbMCralM,1066
-xlin-0.1.35.dist-info/METADATA,sha256=R4tDPREczsJVCRwBejV2IhpeWY-L6nz9aCu1Xyhl4PA,1098
-xlin-0.1.35.dist-info/WHEEL,sha256=IrRNNNJ-uuL1ggO5qMvT1GGhQVdQU54d6ZpYqEZfEWo,92
-xlin-0.1.35.dist-info/RECORD,,
+xlin-0.1.37.dist-info/LICENSE,sha256=60ys6rRtc1dZOP8UjSUr9fAqhZudT3WpKe5WbMCralM,1066
+xlin-0.1.37.dist-info/METADATA,sha256=yMXp-Qqsf8Ch_xEzx4M9y3b21v0zRVsiDAGi4kbKEIo,1098
+xlin-0.1.37.dist-info/WHEEL,sha256=IrRNNNJ-uuL1ggO5qMvT1GGhQVdQU54d6ZpYqEZfEWo,92
+xlin-0.1.37.dist-info/RECORD,,