vizpaint 0.1.0__tar.gz

vizpaint-0.1.0/PKG-INFO

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: vizpaint
+Version: 0.1.0
+Summary: A comprehensive data visualization library
+Author: Your Name
+Requires-Python: >=3.7
+Requires-Dist: matplotlib>=3.3
+Requires-Dist: numpy
+Dynamic: author
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

vizpaint-0.1.0/setup.cfg

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

vizpaint-0.1.0/setup.py

@@ -0,0 +1,11 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="vizpaint", # 包名，在PyPI上必须唯一
+    version="0.1.0",
+    author="Your Name",
+    description="A comprehensive data visualization library",
+    packages=find_packages(),
+    install_requires=["matplotlib>=3.3", "numpy"], # 关键依赖
+    python_requires=">=3.7",
+)

vizpaint-0.1.0/vizpaint/__init__.py

@@ -0,0 +1,106 @@
+
+
+# paint/__init__.py
+""""
+Paint Chart Package - 图表绘制工具包
+提供丰富的图表绘制功能
+"""
+
+__version__ = "0.1.0"
+__author__ = "Chart Package Team"
+
+# 定义所有可导出的函数
+__all__ = [
+    # 基础图表
+    'pie_chart',
+    'bar_chart',
+    'curve_statistical_chart',
+    'scatter_plot',
+    # 新增图表类型
+    'line_plot',
+    'histogram',
+    'box_plot',
+    'area_chart',
+    'radar_chart',
+    'heatmap',
+    'pair_plot',
+    'violin_plot',
+    'bubble_chart',
+    'stacked_bar_chart',
+    # 实用函数
+    'set_style',
+    'show_all',
+    'clear_all',
+    'save_figure',
+    'get_version',
+    'create_subplots',
+    'rose_chart',
+    'surface_3d',
+    'scatter_3d',
+    'stream_plot',
+    'treemap',
+    'quiver_plot'
+]
+
+# 导入基础图表函数
+from .pie_chart import pie_chart
+from .bar_chart import bar_chart
+from .curve_statistical_chart import curve_statistical_chart
+from .scatter_plot import scatter_plot
+
+# 导入新增图表函数
+from .line_plot import line_plot
+from .histogram import histogram
+from .box_plot import box_plot
+from .area_chart import area_chart
+from .radar_chart import radar_chart
+from .heatmap import heatmap
+from .pair_plot import pair_plot
+from .violin_plot import violin_plot
+from .bubble_chart import bubble_chart
+from .stacked_bar_chart import stacked_bar_chart
+from .rose_chart import rose_chart
+from .surface_3d import surface_3d
+from .scatter_3d import scatter_3d
+from .stream_plot import stream_plot
+from .treemap import treemap
+from .quiver_plot import quiver_plot
+
+
+# 导入实用函数
+import matplotlib.pyplot as plt
+import numpy as np
+
+def set_style(style='default'):
+    """设置图表样式"""
+    try:
+        plt.style.use(style)
+        return True
+    except:
+        plt.style.use('default')
+        print(f"样式 '{style}' 不可用，使用默认样式")
+        return False
+
+def show_all():
+    """显示所有图表"""
+    plt.show()
+
+def clear_all():
+    """清除所有图表"""
+    plt.close('all')
+
+def save_figure(filename, dpi=300, transparent=False):
+    """保存当前图表"""
+    plt.savefig(filename, dpi=dpi, transparent=transparent,
+                bbox_inches='tight')
+    plt.close()
+
+def get_version():
+    """获取包版本"""
+    return __version__
+
+def create_subplots(nrows=1, ncols=1, figsize=(10, 8), **kwargs):
+    """创建子图布局"""
+    return plt.subplots(nrows, ncols, figsize=figsize, **kwargs)
+
+# 包初始化消息

vizpaint-0.1.0/vizpaint/area_chart.py

@@ -0,0 +1,64 @@
+# paint/area_chart.py
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+def area_chart(x_data=None, y_data_list=None, labels=None,
+               title="Area Chart", xlabel="X", ylabel="Y",
+               colors=None, alpha=0.5, stacked=False):
+    """
+    绘制面积图
+
+    参数:
+    x_data: X轴数据
+    y_data_list: Y轴数据列表
+    labels: 数据标签
+    title: 图表标题
+    xlabel: X轴标签
+    ylabel: Y轴标签
+    colors: 颜色列表
+    alpha: 透明度
+    stacked: 是否堆叠
+    """
+    fig, ax = plt.subplots(figsize=(10, 6))
+
+    # 生成默认数据
+    if x_data is None:
+        x_data = np.arange(1, 11)
+    if y_data_list is None:
+        y_data_list = [
+            np.random.randint(10, 50, 10),
+            np.random.randint(20, 60, 10),
+            np.random.randint(5, 40, 10)
+        ]
+        labels = ['Product A', 'Product B', 'Product C']
+
+    # 设置默认颜色
+    if colors is None:
+        colors = plt.cm.Pastel1(np.linspace(0, 1, len(y_data_list)))
+
+    # 绘制面积图
+    if stacked:
+        # 堆叠面积图
+        y_stacked = np.vstack(y_data_list)
+        ax.stackplot(x_data, y_stacked, labels=labels,
+                     colors=colors, alpha=alpha)
+    else:
+        # 普通面积图
+        for i, y_data in enumerate(y_data_list):
+            ax.fill_between(x_data, 0, y_data,
+                            alpha=alpha, color=colors[i % len(colors)],
+                            label=labels[i] if labels else None)
+
+    # 设置图表属性
+    ax.set_title(title, fontsize=14, fontweight='bold', pad=20)
+    ax.set_xlabel(xlabel, fontsize=12)
+    ax.set_ylabel(ylabel, fontsize=12)
+
+    ax.grid(True, linestyle='--', alpha=0.3)
+
+    if labels:
+        ax.legend(fontsize=10, loc='best')
+
+    plt.tight_layout()
+    return fig, ax

vizpaint-0.1.0/vizpaint/bar_chart.py

@@ -0,0 +1,32 @@
+# paint/bar_chart.py
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+def bar_chart(a, b, c, d, e, f):
+    """
+    绘制条形图
+
+    参数:
+    a,b,c,d,e: 条形的高度值
+    f: 条形的颜色
+    """
+    n = 6  # 6个条形
+    y = [a, b, c, d, e, 0]  # 你只给了5个值，所以第6个设为0
+    index = np.arange(n)
+
+    plt.figure(figsize=(10, 6))
+    plt.bar(index, y, color=f, edgecolor='black', alpha=0.8)
+
+    # 添加标签
+    plt.xlabel('类别')
+    plt.ylabel('数值')
+    plt.title('条形图')
+    plt.xticks(index, ['A', 'B', 'C', 'D', 'E', 'F'])
+
+    # 在条形上显示数值
+    for i, v in enumerate(y):
+        plt.text(i, v + 0.02, str(v), ha='center', va='bottom')
+
+    plt.tight_layout()
+    plt.show()

vizpaint-0.1.0/vizpaint/box_plot.py

@@ -0,0 +1,72 @@
+# paint/box_plot.py
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+def box_plot(data_list=None, labels=None, title="Box Plot",
+             ylabel="Value", colors=None, show_mean=False,
+             show_points=False, horizontal=False):
+    """
+    绘制箱线图
+
+    参数:
+    data_list: 数据列表（每个元素是一个数据集）
+    labels: 数据集标签
+    title: 图表标题
+    ylabel: Y轴标签
+    colors: 颜色列表
+    show_mean: 是否显示均值
+    show_points: 是否显示数据点
+    horizontal: 是否水平显示
+    """
+    fig, ax = plt.subplots(figsize=(10, 6))
+
+    # 生成默认数据
+    if data_list is None:
+        np.random.seed(42)
+        data_list = [
+            np.random.normal(0, 1, 100),
+            np.random.normal(2, 1.5, 100),
+            np.random.normal(-1, 0.5, 100),
+            np.random.normal(3, 2, 100)
+        ]
+        labels = ['Group A', 'Group B', 'Group C', 'Group D']
+
+    # 设置默认颜色
+    if colors is None:
+        colors = plt.cm.Set2(np.linspace(0, 1, len(data_list)))
+
+    # 绘制箱线图
+    bp = ax.boxplot(data_list, labels=labels, patch_artist=True,
+                    showmeans=show_mean, showfliers=show_points,
+                    vert=not horizontal)
+
+    # 设置箱子颜色
+    for patch, color in zip(bp['boxes'], colors):
+        patch.set_facecolor(color)
+        patch.set_alpha(0.7)
+
+    # 设置其他元素颜色
+    for element in ['whiskers', 'caps', 'medians']:
+        for line in bp[element]:
+            line.set_color('black')
+            line.set_linewidth(1.5)
+
+    # 设置均值点
+    if show_mean:
+        for mean_line in bp['means']:
+            mean_line.set_color('red')
+            mean_line.set_linewidth(2)
+
+    # 设置图表属性
+    ax.set_title(title, fontsize=14, fontweight='bold', pad=20)
+    ax.set_ylabel(ylabel, fontsize=12)
+
+    if horizontal:
+        ax.set_xlabel(ylabel, fontsize=12)
+        ax.set_ylabel('Groups', fontsize=12)
+
+    ax.grid(True, linestyle='--', alpha=0.3, axis='y' if not horizontal else 'x')
+
+    plt.tight_layout()
+    return fig, ax, bp

vizpaint-0.1.0/vizpaint/bubble_chart.py

@@ -0,0 +1,83 @@
+# paint/bubble_chart.py
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+def bubble_chart(x_data=None, y_data=None, size_data=None,
+                 color_data=None, labels=None, title="Bubble Chart",
+                 xlabel="X", ylabel="Y", cmap='viridis',
+                 alpha=0.6, edgecolors='black'):
+    """
+    绘制气泡图
+
+    参数:
+    x_data: X轴数据
+    y_data: Y轴数据
+    size_data: 气泡大小数据
+    color_data: 气泡颜色数据
+    labels: 气泡标签
+    title: 图表标题
+    xlabel: X轴标签
+    ylabel: Y轴标签
+    cmap: 颜色映射
+    alpha: 透明度
+    edgecolors: 边框颜色
+    """
+    fig, ax = plt.subplots(figsize=(10, 8))
+
+    # 生成默认数据
+    if x_data is None:
+        np.random.seed(42)
+        n_points = 30
+        x_data = np.random.rand(n_points) * 100
+        y_data = np.random.rand(n_points) * 100
+        size_data = np.random.rand(n_points) * 1000
+        color_data = np.random.rand(n_points)
+        labels = [f'Point {i + 1}' for i in range(n_points)]
+
+    # 归一化大小数据用于绘图
+    if size_data is not None:
+        size_normalized = (size_data - np.min(size_data)) / (np.max(size_data) - np.min(size_data))
+        sizes = 50 + size_normalized * 500
+    else:
+        sizes = 100
+
+    # 绘制气泡图
+    scatter = ax.scatter(x_data, y_data, s=sizes,
+                         c=color_data, cmap=cmap,
+                         alpha=alpha, edgecolors=edgecolors,
+                         linewidth=1)
+
+    # 添加标签（可选）
+    if labels:
+        for i, label in enumerate(labels):
+            ax.annotate(label, (x_data[i], y_data[i]),
+                        xytext=(5, 5), textcoords='offset points',
+                        fontsize=9, alpha=0.7)
+
+    # 设置图表属性
+    ax.set_title(title, fontsize=14, fontweight='bold', pad=20)
+    ax.set_xlabel(xlabel, fontsize=12)
+    ax.set_ylabel(ylabel, fontsize=12)
+
+    ax.grid(True, linestyle='--', alpha=0.5)
+
+    # 添加颜色条（如果有颜色数据）
+    if color_data is not None:
+        cbar = fig.colorbar(scatter, ax=ax)
+        cbar.set_label('Color Value', fontsize=10)
+
+    # 添加图例表示气泡大小（可选）
+    if size_data is not None:
+        # 创建气泡大小示例
+        min_size = np.min(size_data)
+        max_size = np.max(size_data)
+        median_size = np.median(size_data)
+
+        legend_text = f'Bubble Size\nMin: {min_size:.1f}\nMed: {median_size:.1f}\nMax: {max_size:.1f}'
+        props = dict(boxstyle='round', facecolor='wheat', alpha=0.5)
+        ax.text(0.02, 0.98, legend_text, transform=ax.transAxes, fontsize=9,
+                verticalalignment='top', bbox=props)
+
+    plt.tight_layout()
+    return fig, ax, scatter

vizpaint-0.1.0/vizpaint/curve_statistical_chart.py

@@ -0,0 +1,30 @@
+# paint/curve_statistical_chart.py
+from matplotlib.pyplot import *
+import numpy as np
+
+
+def curve_statistical_chart(a, b, c, d):
+    """
+    绘制曲线统计图
+
+    参数:
+    a: 数据点数量
+    b: 图表标题
+    c: X轴标签
+    d: Y轴标签
+    """
+    x = np.linspace(0, 3 * np.pi, a)
+    y1, y2 = np.sin(x), np.cos(x)
+
+    plot(x, y1, label='sin(x)')
+    plot(x, y2, label='cos(x)')
+    title(b)
+    xlabel(c)
+    ylabel(d)
+    legend()
+    grid(True)
+
+    show()
+
+    # 如果你想让函数返回图表对象以便进一步操作，可以改为：
+    # return plt.gcf(), plt.gca()

vizpaint-0.1.0/vizpaint/heatmap.py

@@ -0,0 +1,60 @@
+# paint/heatmap.py
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+def heatmap(data=None, row_labels=None, col_labels=None,
+            title="Heatmap", cmap='viridis', annotate=True,
+            fmt='.2f', cbar=True, square=False):
+    """
+    绘制热力图
+
+    参数:
+    data: 二维数据数组
+    row_labels: 行标签
+    col_labels: 列标签
+    title: 图表标题
+    cmap: 颜色映射
+    annotate: 是否显示数值
+    fmt: 数值格式
+    cbar: 是否显示颜色条
+    square: 是否保持方形
+    """
+    fig, ax = plt.subplots(figsize=(10, 8))
+
+    # 生成默认数据
+    if data is None:
+        np.random.seed(42)
+        data = np.random.rand(8, 10)
+        row_labels = [f'Row {i + 1}' for i in range(8)]
+        col_labels = [f'Col {i + 1}' for i in range(10)]
+
+    # 绘制热力图
+    im = ax.imshow(data, cmap=cmap, aspect='auto' if not square else 'equal')
+
+    # 添加数值标签
+    if annotate:
+        for i in range(data.shape[0]):
+            for j in range(data.shape[1]):
+                text = ax.text(j, i, format(data[i, j], fmt),
+                               ha="center", va="center",
+                               color="white" if data[i, j] > 0.6 else "black",
+                               fontsize=8)
+
+    # 设置刻度标签
+    if row_labels is not None:
+        ax.set_yticks(np.arange(data.shape[0]))
+        ax.set_yticklabels(row_labels)
+    if col_labels is not None:
+        ax.set_xticks(np.arange(data.shape[1]))
+        ax.set_xticklabels(col_labels, rotation=45, ha='right')
+
+    # 设置标题
+    ax.set_title(title, fontsize=14, fontweight='bold', pad=20)
+
+    # 添加颜色条
+    if cbar:
+        fig.colorbar(im, ax=ax, fraction=0.046, pad=0.04)
+
+    plt.tight_layout()
+    return fig, ax, im

vizpaint-0.1.0/vizpaint/histogram.py

@@ -0,0 +1,79 @@
+# paint/histogram.py
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+def histogram(data=None, bins=10, title="Histogram",
+              xlabel="Value", ylabel="Frequency",
+              color='skyblue', edgecolor='black',
+              density=False, cumulative=False,
+              show_mean=False, show_median=False):
+    """
+    绘制直方图
+
+    参数:
+    data: 数据数组
+    bins: 柱子数量或边界
+    title: 图表标题
+    xlabel: X轴标签
+    ylabel: Y轴标签
+    color: 柱子颜色
+    edgecolor: 边框颜色
+    density: 是否显示密度
+    cumulative: 是否累积
+    show_mean: 是否显示均值线
+    show_median: 是否显示中位数线
+    """
+    fig, ax = plt.subplots(figsize=(10, 6))
+
+    # 生成默认数据
+    if data is None:
+        np.random.seed(42)
+        data = np.random.normal(0, 1, 1000)
+
+    # 绘制直方图
+    n, bins, patches = ax.hist(data, bins=bins,
+                               color=color, edgecolor=edgecolor,
+                               alpha=0.7, density=density,
+                               cumulative=cumulative)
+
+    # 计算统计量
+    mean_val = np.mean(data)
+    median_val = np.median(data)
+    std_val = np.std(data)
+
+    # 添加统计线
+    if show_mean:
+        ax.axvline(mean_val, color='red', linestyle='--',
+                   linewidth=2, label=f'Mean: {mean_val:.2f}')
+
+    if show_median:
+        ax.axvline(median_val, color='green', linestyle='-.',
+                   linewidth=2, label=f'Median: {median_val:.2f}')
+
+    # 添加正态分布曲线（如果适用）
+    if density and not cumulative:
+        from scipy.stats import norm
+        xmin, xmax = ax.get_xlim()
+        x = np.linspace(xmin, xmax, 100)
+        p = norm.pdf(x, mean_val, std_val)
+        ax.plot(x, p, 'k', linewidth=2, label='Normal Distribution')
+
+    # 设置图表属性
+    ax.set_title(title, fontsize=14, fontweight='bold', pad=20)
+    ax.set_xlabel(xlabel, fontsize=12)
+    ax.set_ylabel('Density' if density else 'Frequency', fontsize=12)
+
+    ax.grid(True, linestyle='--', alpha=0.3)
+
+    if show_mean or show_median:
+        ax.legend(fontsize=10)
+
+    # 添加统计信息文本框
+    textstr = f'N = {len(data)}\nMean = {mean_val:.2f}\nStd = {std_val:.2f}'
+    props = dict(boxstyle='round', facecolor='wheat', alpha=0.5)
+    ax.text(0.02, 0.98, textstr, transform=ax.transAxes, fontsize=10,
+            verticalalignment='top', bbox=props)
+
+    plt.tight_layout()
+    return fig, ax, (n, bins, patches)

vizpaint-0.1.0/vizpaint/line_plot.py

@@ -0,0 +1,70 @@
+# paint/line_plot.py
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+def line_plot(x_data=None, y_data_list=None, labels=None,
+              title="Line Plot", xlabel="X", ylabel="Y",
+              line_styles=None, colors=None, marker_styles=None,
+              grid=True, legend=True):
+    """
+    绘制折线图
+
+    参数:
+    x_data: X轴数据
+    y_data_list: Y轴数据列表（多条线）
+    labels: 图例标签
+    title: 图表标题
+    xlabel: X轴标签
+    ylabel: Y轴标签
+    line_styles: 线条样式列表
+    colors: 颜色列表
+    marker_styles: 标记样式列表
+    grid: 是否显示网格
+    legend: 是否显示图例
+    """
+    fig, ax = plt.subplots(figsize=(10, 6))
+
+    # 默认数据
+    if x_data is None:
+        x_data = np.arange(1, 11)
+    if y_data_list is None:
+        y_data_list = [np.sin(x_data), np.cos(x_data), np.tan(x_data) / 10]
+        labels = ['sin(x)', 'cos(x)', 'tan(x)/10']
+
+    # 设置默认样式
+    if colors is None:
+        colors = plt.cm.Set1(np.linspace(0, 1, len(y_data_list)))
+    if line_styles is None:
+        line_styles = ['-', '--', '-.', ':'] * (len(y_data_list) // 4 + 1)
+    if marker_styles is None:
+        marker_styles = ['o', 's', '^', 'D', 'v'] * (len(y_data_list) // 5 + 1)
+    if labels is None:
+        labels = [f'Line {i + 1}' for i in range(len(y_data_list))]
+
+    # 绘制每条线
+    lines = []
+    for i, y_data in enumerate(y_data_list):
+        line, = ax.plot(x_data, y_data,
+                        linestyle=line_styles[i % len(line_styles)],
+                        color=colors[i % len(colors)],
+                        marker=marker_styles[i % len(marker_styles)],
+                        markersize=8,
+                        linewidth=2,
+                        label=labels[i],
+                        alpha=0.8)
+        lines.append(line)
+
+    # 设置图表属性
+    ax.set_title(title, fontsize=14, fontweight='bold', pad=20)
+    ax.set_xlabel(xlabel, fontsize=12)
+    ax.set_ylabel(ylabel, fontsize=12)
+
+    if grid:
+        ax.grid(True, linestyle='--', alpha=0.5)
+
+    if legend and len(y_data_list) > 1:
+        ax.legend(fontsize=10, loc='best')
+
+    plt.tight_layout()
+    return fig, ax, lines

vizpaint-0.1.0/vizpaint/paintchart.py

@@ -0,0 +1,33 @@
+from matplotlib.pyplot import *
+from numpy import *
+#Import required modules
+
+def pie_chart(a,b,c,d,e,f,g,h,i,j,k,l):
+    #Draw a pie chart
+    labels = a,b,c,d,e,f
+    sizes = [g,h,i,j,k,l]
+    colors = ['green','blue','red','yellow','purple','orange']
+    pie(sizes, labels=labels, colors= colors, autopct ='%1.1f%%', shadow=True, startangle=90)
+    axis('equal')
+    show()
+def bar_chart(a,b,c,d,e,f):
+    #Draw a bar chart
+    n = 6
+    y = [a,b,c,d,e]
+    index = arange(n)
+    bar(left = index, height= y, cocor = f, )
+def curve_statistical_chart(a,b,c,d):
+    # Draw a curve statistical chart
+    x = linspace(0,3 * np.pi ,a )
+    y1,y2 = sin(x),cos(x)
+    plot(x, y1)
+    plot(x, y2)
+    title(b)
+    xlabel(c)
+    ylabel(d)
+    show()
+def scatter_plot(a,b,c,d,e,f,g,h):
+    #Draw a scatter plot
+    higth = [a,b,c,d]
+    weiggt= [e,f,g,h]
+    scatter(higth,weiggt)