oafuncs 0.0.80__py2.py3-none-any.whl → 0.0.82__py2.py3-none-any.whl

oafuncs/oa_down/test.py

@@ -1,151 +0,0 @@
-#!/usr/bin/env python
-# coding=utf-8
-"""
-Author: Liu Kun && 16031215@qq.com
-Date: 2024-12-01 19:32:25
-LastEditors: Liu Kun && 16031215@qq.com
-LastEditTime: 2024-12-10 11:16:36
-FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_down\\test.py
-Description:
-EditPlatform: vscode
-ComputerInfo: XPS 15 9510
-SystemInfo: Windows 11
-Python Version: 3.12
-"""
-
-import os
-import random
-import re
-
-
-def is_valid_user_agent(user_agent):
-    # 简单的正则表达式来检查User Agent的格式
-    # 这个正则表达式检查User Agent是否包含常见的浏览器信息格式
-    pattern = re.compile(
-        r"^(?:(?:Mozilla|Opera|Chrome|Safari|Edg|OPR)/[\d.]+)"
-        r"(?:\s(?:\(.*?\)))?"
-        r"(?:\s(?:Gecko|AppleWebKit|KHTML, like Gecko|Version|Edge|OPR)/[\d.]+)?"
-        r"(?:\s.*?(?:rv:|Version/|Ubuntu|Macintosh|Windows|X11|Linux|CrOS|FreeBSD|OpenBSD|NetBSD|iPhone|iPad|iPod|Android|BlackBerry|BB10|Mobile|Symbian|Windows Phone|IEMobile|Opera Mini|Opera Mobi|UCBrowser|MQQBrowser|baiduboxapp|baidubrowser|Safari|Firefox|MSIE|Trident|Edge|EdgA|Chrome|CriOS|Vivaldi|Sleipnir|Midori|ELinks|Lynx|w3m|Arora|Epiphany|Konqueror|Dillo|Netscape|SeaMonkey|K-Meleon|Camino|Iceape|Galeon|GranParadiso|Iceweasel|Firefox|Fennec|Conkeror|PaleMoon|Uzbl|QupZilla|Otter|Waterfox|Basilisk|Cyberfox|PaleMoon|GNU IceCat|GNU IceWeasel|IceCat|IceWeasel|Seamonkey|Iceape|Firefox|Epiphany|Web|Safari|Android|Mobile|BlackBerry|BB10|Tablet|Silk|Kindle|FxiOS|Focus|SamsungBrowser|browser|AppleWebKit|Puffin|DuckDuckGo|YaBrowser|Yandex|Amigo|NokiaBrowser|OviBrowser|OneBrowser|Chrome|Firefox|Safari|OPR|Coast|Mercury|Silk|Skyfire|IEMobile|Bolt|Jasmine|NativeHost|Crosswalk|TizenBrowser|SailfishBrowser|SamsungBrowser|Silk-Accelerated|UCBrowser|Quark|XiaoMi|OnePlus|Vivo|Oppo|Realme|Meizu|Lenovo|Huawei|ZTE|Alcatel|Sony|Nokia|LG|HTC|Asus|Acer|Motorola|Samsung)/[\d.]+)?$"
-    )
-
-    # 使用正则表达式匹配User Agent字符串
-    if pattern.match(user_agent):
-        return True
-    else:
-        return False
-
-
-def get_ua():
-    current_dir = os.path.dirname(os.path.abspath(__file__))
-    ua_file_txt = os.path.join(current_dir, "User_Agent-list.txt")
-
-    with open(ua_file_txt, "r") as f:
-        ua_list = f.readlines()
-        # 去掉换行符和空行
-        ua_list = [line.strip() for line in ua_list if line.strip()]
-
-    return random.choice(ua_list)
-
-
-def get_ua_org():
-    ua_list = [
-        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537.36 OPR/26.0.1656.60",
-        "Opera/8.0 (Windows NT 5.1; U; en)",
-        "Mozilla/5.0 (Windows NT 5.1; U; en; rv:1.8.1) Gecko/20061208 Firefox/2.0.0 Opera 9.50",
-        "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; en) Opera 9.50",
-        "Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; en) Presto/2.8.131 Version/11.11",
-        "Opera/9.80 (Windows NT 6.1; U; en) Presto/2.8.131 Version/11.11",
-        "Opera/9.80 (Android 2.3.4; Linux; Opera Mobi/build-1107180945; U; en-GB) Presto/2.8.149 Version/11.10",
-        "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:34.0) Gecko/20100101 Firefox/34.0",
-        "Mozilla/5.0 (X11; U; Linux x86_64; zh-CN; rv:1.9.2.10) Gecko/20100922 Ubuntu/10.10 (maverick) Firefox/3.6.10",
-        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.6; rv,2.0.1) Gecko/20100101 Firefox/4.0.1",
-        "Mozilla/5.0 (Windows NT 6.1; rv,2.0.1) Gecko/20100101 Firefox/4.0.1",
-        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/534.57.2 (KHTML, like Gecko) Version/5.1.7 Safari/534.57.2",
-        "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.122 Safari/537.36",
-        "Mozilla/5.0 (Windows; U; Windows NT 6.1; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
-        "Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
-        "Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
-        "Mozilla/5.0 (iPad; U; CPU OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
-        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.71 Safari/537.36",
-        "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.11 (KHTML, like Gecko) Chrome/23.0.1271.64 Safari/537.11",
-        "Mozilla/5.0 (Windows; U; Windows NT 6.1; en-US) AppleWebKit/534.16 (KHTML, like Gecko) Chrome/10.0.648.133 Safari/534.16",
-        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_0) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
-        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/30.0.1599.101 Safari/537.36",
-        "Mozilla/5.0 (Windows NT 6.1; WOW64; Trident/7.0; rv:11.0) like Gecko",
-        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; 360SE)",
-        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/536.11 (KHTML, like Gecko) Chrome/20.0.1132.11 TaoBrowser/2.0 Safari/536.11",
-        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.71 Safari/537.1 LBBROWSER",
-        "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; WOW64; Trident/5.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; .NET4.0E; LBBROWSER)",
-        "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; QQDownload 732; .NET4.0C; .NET4.0E; LBBROWSER)",
-        "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; WOW64; Trident/5.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; .NET4.0E; QQBrowser/7.0.3698.400)",
-        "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; QQDownload 732; .NET4.0C; .NET4.0E)",
-        "Mozilla/5.0 (Windows NT 5.1) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.84 Safari/535.11 SE 2.X MetaSr 1.0",
-        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SV1; QQDownload 732; .NET4.0C; .NET4.0E; SE 2.X MetaSr 1.0)",
-        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SE 2.X MetaSr 1.0; SE 2.X MetaSr 1.0; .NET CLR 2.0.50727; SE 2.X MetaSr 1.0)",
-        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Maxthon/4.4.3.4000 Chrome/30.0.1599.101 Safari/537.36",
-        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_0) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
-        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/38.0.2125.122 UBrowser/4.0.3214.0 Safari/537.36",
-        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/55.0.2883.87 UBrowser/6.2.4094.1 Safari/537.36",
-        "Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
-        "Mozilla/5.0 (iPod; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
-        "Mozilla/5.0 (iPad; U; CPU OS 4_2_1 like Mac OS X; zh-cn) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8C148 Safari/6533.18.5",
-        "Mozilla/5.0 (iPad; U; CPU OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
-        "Mozilla/5.0 (Linux; U; Android 2.2.1; zh-cn; HTC_Wildfire_A3333 Build/FRG83D) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
-        "Mozilla/5.0 (Linux; U; Android 2.3.7; en-us; Nexus One Build/FRF91) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
-        "MQQBrowser/26 Mozilla/5.0 (Linux; U; Android 2.3.7; zh-cn; MB200 Build/GRJ22; CyanogenMod-7) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
-        "Opera/9.80 (Android 2.3.4; Linux; Opera Mobi/build-1107180945; U; en-GB) Presto/2.8.149 Version/11.10",
-        "Mozilla/5.0 (Linux; U; Android 3.0; en-us; Xoom Build/HRI39) AppleWebKit/534.13 (KHTML, like Gecko) Version/4.0 Safari/534.13",
-        "Mozilla/5.0 (BlackBerry; U; BlackBerry 9800; en) AppleWebKit/534.1+ (KHTML, like Gecko) Version/6.0.0.337 Mobile Safari/534.1+",
-        "Mozilla/5.0 (hp-tablet; Linux; hpwOS/3.0.0; U; en-US) AppleWebKit/534.6 (KHTML, like Gecko) wOSBrowser/233.70 Safari/534.6 TouchPad/1.0",
-        "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0;",
-        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0)",
-        "Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0)",
-        "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)",
-        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
-        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; The World)",
-        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; TencentTraveler 4.0)",
-        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Avant Browser)",
-        "Mozilla/5.0 (Linux; U; Android 2.3.7; en-us; Nexus One Build/FRF91) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
-        "Mozilla/5.0 (SymbianOS/9.4; Series60/5.0 NokiaN97-1/20.0.019; Profile/MIDP-2.1 Configuration/CLDC-1.1) AppleWebKit/525 (KHTML, like Gecko) BrowserNG/7.1.18124",
-        "Mozilla/5.0 (compatible; MSIE 9.0; Windows Phone OS 7.5; Trident/5.0; IEMobile/9.0; HTC; Titan)",
-        "UCWEB7.0.2.37/28/999",
-        "NOKIA5700/UCWEB7.0.2.37/28/999",
-        "Openwave/UCWEB7.0.2.37/28/999",
-        "Openwave/UCWEB7.0.2.37/28/999",
-    ]
-    with open(newtxtfile, "w") as f:
-        for line in ua_list:
-            f.write(line + "\n")
-    # print(f'Using User-Agent: {ua}')
-    ua = random.choice(ua_list)
-    return ua
-
-
-# get_ua_org()
-
-if __name__ == "__main__":
-    txtfile = r"E:\Code\Python\My_Funcs\OAFuncs\oafuncs\oa_down\User_Agent-list.txt"
-
-    with open(txtfile, "r") as f:
-        lines = f.readlines()
-        # 去掉换行符和空行
-        lines = [line.strip() for line in lines if line.strip()]
-    """ new_line = []
-    for i in range(len(lines)):
-        if '/' in lines[i]:
-            new_line.append(lines[i])
-        else:
-            print(lines[i]) """
-
-    new_line = []
-    for line in lines:
-        if is_valid_user_agent(line):
-            # print(line)
-            new_line.append(line)
-        else:
-            print(f"Invalid User-Agent: {line}")
-
-    newtxtfile = r"E:\Code\Python\My_Funcs\OAFuncs\oafuncs\oa_down\ua_list_new.txt"
-    with open(newtxtfile, "w") as f:
-        for line in new_line:
-            f.write(line + "\n")

oafuncs/oa_s/__init__.py

@@ -1,23 +0,0 @@
-#!/usr/bin/env python
-# coding=utf-8
-"""
-Author: Liu Kun && 16031215@qq.com
-Date: 2024-09-17 16:09:20
-LastEditors: Liu Kun && 16031215@qq.com
-LastEditTime: 2024-12-13 10:47:40
-FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_s\\__init__.py
-Description:
-EditPlatform: vscode
-ComputerInfo: XPS 15 9510
-SystemInfo: Windows 11
-Python Version: 3.11
-"""
-
-# 会导致OAFuncs直接导入所有函数，不符合模块化设计
-from .oa_cmap import *
-from .oa_data import *
-from .oa_draw import *
-from .oa_file import *
-from .oa_help import *
-from .oa_nc import *
-from .oa_python import *

oafuncs/oa_s/oa_cmap.py

@@ -1,163 +0,0 @@
-#!/usr/bin/env python
-# coding=utf-8
-"""
-Author: Liu Kun && 16031215@qq.com
-Date: 2024-09-17 16:55:11
-LastEditors: Liu Kun && 16031215@qq.com
-LastEditTime: 2024-11-21 13:14:24
-FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_cmap.py
-Description:
-EditPlatform: vscode
-ComputerInfo: XPS 15 9510
-SystemInfo: Windows 11
-Python Version: 3.11
-"""
-
-import matplotlib as mpl
-import matplotlib.pyplot as plt
-import numpy as np
-
-__all__ = ["show", "cmap2colors", "create_cmap", "create_cmap_rgbtxt", "choose_cmap"]
-
-# ** 将cmap用填色图可视化（官网摘抄函数）
-def show(colormaps: list):
-    """
-    Helper function to plot data with associated colormap.
-    example:
-    cmap = ListedColormap(["darkorange", "gold", "lawngreen", "lightseagreen"])
-    show([cmap])
-    """
-    np.random.seed(19680801)
-    data = np.random.randn(30, 30)
-    n = len(colormaps)
-    fig, axs = plt.subplots(1, n, figsize=(n * 2 + 2, 3), constrained_layout=True, squeeze=False)
-    for [ax, cmap] in zip(axs.flat, colormaps):
-        psm = ax.pcolormesh(data, cmap=cmap, rasterized=True, vmin=-4, vmax=4)
-        fig.colorbar(psm, ax=ax)
-    plt.show()
-
-
-# ** 将cmap转为list，即多个颜色的列表
-def cmap2colors(cmap, n=256):
-    """
-    cmap        : cmap名称
-    n           : 提取颜色数量
-    return      : 提取的颜色列表
-    example     : out_colors = cmap2colors('viridis', 256)
-    """
-    c_map = mpl.colormaps.get_cmap(cmap)
-    out_colors = [c_map(i) for i in np.linspace(0, 1, n)]
-    return out_colors
-
-
-# ** 自制cmap，多色，可带位置
-def create_cmap(colors: list, nodes=None, under=None, over=None):  # 利用颜色快速配色
-    """
-    func        : 自制cmap，自动确定颜色位置（等比例）
-    description : colors可以是颜色名称，也可以是十六进制颜色代码
-    param        {*} colors 颜色
-    param        {*} nodes 颜色位置，默认不提供，等间距
-    return       {*} cmap
-    example     : cmap = create_cmap(['#C2B7F3','#B3BBF2','#B0CBF1','#ACDCF0','#A8EEED'])
-                cmap = create_cmap(['aliceblue','skyblue','deepskyblue'],[0.0,0.5,1.0])
-    """
-    if nodes is None:  # 采取自动分配比例
-        cmap_color = mpl.colors.LinearSegmentedColormap.from_list("mycmap", colors)
-    else:  # 按照提供比例分配
-        cmap_color = mpl.colors.LinearSegmentedColormap.from_list("mycmap", list(zip(nodes, colors)))
-    if under is not None:
-        cmap_color.set_under(under)
-    if over is not None:
-        cmap_color.set_over(over)
-    return cmap_color
-
-
-# ** 根据RGB的txt文档制作色卡（利用Grads调色盘）
-def create_cmap_rgbtxt(rgbtxt_file,split_mark=','):  # 根据RGB的txt文档制作色卡/根据rgb值制作
-    """
-    func        : 根据RGB的txt文档制作色卡
-    description : rgbtxt_file='E:/python/colorbar/test.txt'
-    param        {*} rgbtxt_file txt文件路径
-    return       {*} camp
-    example     : cmap=create_cmap_rgbtxt(path,split_mark=',') #
-
-    txt example : 251,251,253
-                225,125,25
-                250,205,255
-    """
-    with open(rgbtxt_file) as fid:
-        data = fid.readlines()
-    n = len(data)
-    rgb = np.zeros((n, 3))
-    for i in np.arange(n):
-        rgb[i][0] = data[i].split(split_mark)[0]
-        rgb[i][1] = data[i].split(split_mark)[1]
-        rgb[i][2] = data[i].split(split_mark)[2]
-    max_rgb = np.max(rgb)
-    if max_rgb > 2:  # 如果rgb值大于2，则认为是0-255的值，需要归一化
-        rgb = rgb / 255.0
-    icmap = mpl.colors.ListedColormap(rgb, name="my_color")
-    return icmap
-
-
-def choose_cmap(cmap_name=None, query=False):
-    """
-    description: Choosing a colormap from the list of available colormaps or a custom colormap
-    param {*} cmap_name:
-    param {*} query:
-    return {*}
-    """
-
-    my_cmap_dict = {
-        "diverging_1": create_cmap(["#4e00b3", "#0000FF", "#00c0ff", "#a1d3ff", "#DCDCDC", "#FFD39B", "#FF8247", "#FF0000", "#FF5F9E"]),
-        "cold_1": create_cmap(["#4e00b3", "#0000FF", "#00c0ff", "#a1d3ff", "#DCDCDC"]),
-        "warm_1": create_cmap(["#DCDCDC", "#FFD39B", "#FF8247", "#FF0000", "#FF5F9E"]),
-        # "land_1": create_custom(["#3E6436", "#678A59", "#91A176", "#B8A87D", "#D9CBB2"], under="#A6CEE3", over="#FFFFFF"), # 陆地颜色从深绿到浅棕，表示从植被到沙地的递减
-        # "ocean_1": create_custom(["#126697", "#2D88B3", "#4EA1C9", "#78B9D8", "#A6CEE3"], under="#8470FF", over="#3E6436"), # 海洋颜色从深蓝到浅蓝，表示从深海到浅海的递减
-        # "ocean_land_1": create_custom(
-        #     [
-        #         "#126697",  # 深蓝（深海）
-        #         "#2D88B3",  # 蓝
-        #         "#4EA1C9",  # 蓝绿
-        #         "#78B9D8",  # 浅蓝（浅海）
-        #         "#A6CEE3",  # 浅蓝（近岸）
-        #         "#AAAAAA",  # 灰色（0值，海平面）
-        #         "#D9CBB2",  # 沙质土壤色（陆地开始）
-        #         "#B8A87D",  # 浅棕
-        #         "#91A176",  # 浅绿
-        #         "#678A59",  # 中绿
-        #         "#3E6436",  # 深绿（高山）
-        #     ]
-        # ),
-        "colorful_1": create_cmap(["#6d00db", "#9800cb", "#F2003C", "#ff4500", "#ff7f00", "#FE28A2", "#FFC0CB", "#DDA0DD", "#40E0D0", "#1a66f2", "#00f7fb", "#8fff88", "#E3FF00"]),
-    }
-    if query:
-        for key, _ in my_cmap_dict.items():
-            print(key)
-
-    if cmap_name in my_cmap_dict:
-        return my_cmap_dict[cmap_name]
-    else:
-        try:
-            return mpl.colormaps.get_cmap(cmap_name)
-        except ValueError:
-            raise ValueError(f"Unknown cmap name: {cmap_name}")
-
-
-if __name__ == "__main__":
-    # ** 测试自制cmap
-    colors = ["#C2B7F3", "#B3BBF2", "#B0CBF1", "#ACDCF0", "#A8EEED"]
-    nodes = [0.0, 0.2, 0.4, 0.6, 1.0]
-    c_map = create_cmap(colors, nodes)
-    show([c_map])
-
-    # ** 测试自制diverging型cmap
-    diverging_cmap = create_cmap(["#4e00b3", "#0000FF", "#00c0ff", "#a1d3ff", "#DCDCDC", "#FFD39B", "#FF8247", "#FF0000", "#FF5F9E"])
-    show([diverging_cmap])
-
-    # ** 测试根据RGB的txt文档制作色卡
-    file_path = "E:/python/colorbar/test.txt"
-    cmap_rgb = create_cmap_rgbtxt(file_path)
-
-    # ** 测试将cmap转为list
-    out_colors = cmap2colors("viridis", 256)

oafuncs/oa_s/oa_data.py

@@ -1,187 +0,0 @@
-#!/usr/bin/env python
-# coding=utf-8
-'''
-Author: Liu Kun && 16031215@qq.com
-Date: 2024-09-17 17:12:47
-LastEditors: Liu Kun && 16031215@qq.com
-LastEditTime: 2024-11-21 13:13:20
-FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_data.py
-Description:  
-EditPlatform: vscode
-ComputerInfo: XPS 15 9510
-SystemInfo: Windows 11
-Python Version: 3.11
-'''
-
-
-import multiprocessing as mp
-from concurrent.futures import ThreadPoolExecutor
-
-import numpy as np
-from scipy.interpolate import griddata
-
-__all__ = ['interp_2d', 'interp_2d_parallel']
-
-# ** 高维插值函数，插值最后两个维度
-
-
-def interp_2d(target_x, target_y, origin_x, origin_y, data, method='linear'):
-    """
-    高维插值函数，默认插值最后两个维度，传输数据前请确保数据的维度正确
-    参数:
-    target_y (array-like): 目标经度网格 1D 或 2D
-    target_x (array-like): 目标纬度网格 1D 或 2D
-    origin_y (array-like): 初始经度网格 1D 或 2D
-    origin_x (array-like): 初始纬度网格 1D 或 2D
-    data (array-like): 数据 (*, lat, lon) 2D, 3D, 4D
-    method (str, optional): 插值方法，可选 'linear', 'nearest', 'cubic' 等，默认为 'linear'
-    返回:
-    array-like: 插值结果
-    """
-
-    # 确保目标网格和初始网格都是二维的
-    if len(target_y.shape) == 1:
-        target_x, target_y = np.meshgrid(target_x, target_y)
-    if len(origin_y.shape) == 1:
-        origin_x, origin_y = np.meshgrid(origin_x, origin_y)
-
-    dims = data.shape
-    len_dims = len(dims)
-    # print(dims[-2:])
-    # 根据经纬度网格判断输入数据的形状是否匹配
-
-    if origin_x.shape != dims[-2:] or origin_y.shape != dims[-2:]:
-        print(origin_x.shape, dims[-2:])
-        raise ValueError('Shape of data does not match shape of origin_x or origin_y.')
-
-    # 将目标网格展平成一维数组
-    target_points = np.column_stack((np.ravel(target_y), np.ravel(target_x)))
-
-    # 将初始网格展平成一维数组
-    origin_points = np.column_stack((np.ravel(origin_y), np.ravel(origin_x)))
-
-    # 进行插值
-    if len_dims == 2:
-        interpolated_data = griddata(origin_points, np.ravel(data), target_points, method=method)
-        interpolated_data = np.reshape(interpolated_data, target_y.shape)
-    elif len_dims == 3:
-        interpolated_data = []
-        for i in range(dims[0]):
-            dt = griddata(origin_points, np.ravel(data[i, :, :]), target_points, method=method)
-            interpolated_data.append(np.reshape(dt, target_y.shape))
-            print(f'Interpolating {i+1}/{dims[0]}...')
-        interpolated_data = np.array(interpolated_data)
-    elif len_dims == 4:
-        interpolated_data = []
-        for i in range(dims[0]):
-            interpolated_data.append([])
-            for j in range(dims[1]):
-                dt = griddata(origin_points, np.ravel(data[i, j, :, :]), target_points, method=method)
-                interpolated_data[i].append(np.reshape(dt, target_y.shape))
-                print(f'\rInterpolating {i*dims[1]+j+1}/{dims[0]*dims[1]}...', end='')
-        print('\n')
-        interpolated_data = np.array(interpolated_data)
-
-    return interpolated_data
-
-
-# ** 高维插值函数，插值最后两个维度，使用多线程进行插值
-# 在本地电脑上可以提速三倍左右，超算上暂时无法加速
-def interp_2d_parallel(target_x, target_y, origin_x, origin_y, data, method='linear'):
-    '''
-    param        {*} target_x 目标经度网格 1D 或 2D
-    param        {*} target_y 目标纬度网格 1D 或 2D
-    param        {*} origin_x 初始经度网格 1D 或 2D
-    param        {*} origin_y 初始纬度网格 1D 或 2D
-    param        {*} data 数据 (*, lat, lon) 2D, 3D, 4D
-    param        {*} method 插值方法，可选 'linear', 'nearest', 'cubic' 等，默认为 'linear'
-    return       {*} 插值结果
-    description : 高维插值函数，默认插值最后两个维度，传输数据前请确保数据的维度正确
-    example     : interpolated_data = interp_2d_parallel(target_x, target_y, origin_x, origin_y, data, method='linear')
-    '''
-    def interp_single2d(target_y, target_x, origin_y, origin_x, data, method='linear'):
-        target_points = np.column_stack((np.ravel(target_y), np.ravel(target_x)))
-        origin_points = np.column_stack((np.ravel(origin_y), np.ravel(origin_x)))
-
-        dt = griddata(origin_points, np.ravel(data[:, :]), target_points, method=method)
-        return np.reshape(dt, target_y.shape)
-
-    def interp_single3d(i, target_y, target_x, origin_y, origin_x, data, method='linear'):
-        target_points = np.column_stack((np.ravel(target_y), np.ravel(target_x)))
-        origin_points = np.column_stack((np.ravel(origin_y), np.ravel(origin_x)))
-
-        dt = griddata(origin_points, np.ravel(data[i, :, :]), target_points, method=method)
-        return np.reshape(dt, target_y.shape)
-
-    def interp_single4d(i, j, target_y, target_x, origin_y, origin_x, data, method='linear'):
-        target_points = np.column_stack((np.ravel(target_y), np.ravel(target_x)))
-        origin_points = np.column_stack((np.ravel(origin_y), np.ravel(origin_x)))
-
-        dt = griddata(origin_points, np.ravel(data[i, j, :, :]), target_points, method=method)
-        return np.reshape(dt, target_y.shape)
-
-    if len(target_y.shape) == 1:
-        target_x, target_y = np.meshgrid(target_x, target_y)
-    if len(origin_y.shape) == 1:
-        origin_x, origin_y = np.meshgrid(origin_x, origin_y)
-
-    dims = data.shape
-    len_dims = len(dims)
-
-    if origin_x.shape != dims[-2:] or origin_y.shape != dims[-2:]:
-        raise ValueError('数据形状与 origin_x 或 origin_y 的形状不匹配.')
-
-    interpolated_data = []
-
-    # 使用多线程进行插值
-    with ThreadPoolExecutor(max_workers=mp.cpu_count()-2) as executor:
-        print(f'Using {mp.cpu_count()-2} threads...')
-        if len_dims == 2:
-            interpolated_data = list(executor.map(interp_single2d, [target_y], [target_x], [origin_y], [origin_x], [data], [method]))
-        elif len_dims == 3:
-            interpolated_data = list(executor.map(interp_single3d, [i for i in range(dims[0])], [target_y]*dims[0], [target_x]*dims[0], [origin_y]*dims[0], [origin_x]*dims[0], [data]*dims[0], [method]*dims[0]))
-        elif len_dims == 4:
-            interpolated_data = list(executor.map(interp_single4d, [i for i in range(dims[0]) for j in range(dims[1])], [j for i in range(dims[0]) for j in range(dims[1])], [target_y]*dims[0]*dims[1], [target_x]*dims[0]*dims[1], [origin_y]*dims[0]*dims[1], [origin_x]*dims[0]*dims[1], [data]*dims[0]*dims[1], [method]*dims[0]*dims[1]))
-            interpolated_data = np.array(interpolated_data).reshape(dims[0], dims[1], target_y.shape[0], target_x.shape[1])
-
-    interpolated_data = np.array(interpolated_data)
-
-    return interpolated_data
-
-
-if __name__ == '__main__':
-    import time
-
-    import matplotlib.pyplot as plt
-
-    # 测试数据
-    origin_x = np.linspace(0, 10, 11)
-    origin_y = np.linspace(0, 10, 11)
-    target_x = np.linspace(0, 10, 101)
-    target_y = np.linspace(0, 10, 101)
-    data = np.random.rand(11, 11)
-
-    # 高维插值
-    origin_x = np.linspace(0, 10, 11)
-    origin_y = np.linspace(0, 10, 11)
-    target_x = np.linspace(0, 10, 101)
-    target_y = np.linspace(0, 10, 101)
-    data = np.random.rand(10, 10, 11, 11)
-
-    start = time.time()
-    interpolated_data = interp_2d(target_x, target_y, origin_x, origin_y, data)
-    print(f'Interpolation time: {time.time()-start:.2f}s')
-
-    print(interpolated_data.shape)
-
-    # 高维插值多线程
-    start = time.time()
-    interpolated_data = interp_2d_parallel(target_x, target_y, origin_x, origin_y, data)
-    print(f'Interpolation time: {time.time()-start:.2f}s')
-
-    print(interpolated_data.shape)
-    print(interpolated_data[0, 0, :, :].shape)
-    plt.figure()
-    plt.contourf(target_x, target_y, interpolated_data[0, 0, :, :])
-    plt.colorbar()
-    plt.show()