byzh-core 0.0.9.2__tar.gz → 0.0.9.4__tar.gz

{byzh_core-0.0.9.2 → byzh_core-0.0.9.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: byzh-core
-Version: 0.0.9.2
+Version: 0.0.9.4
 Summary: byzh-core是byzh系列的核心库，包含了一些常用的工具函数和类。
 Author: byzh_rc
 License: MIT

{byzh_core-0.0.9.2 → byzh_core-0.0.9.4}/byzh/core/Brecorder/record2d.py

@@ -2,10 +2,12 @@ import pandas as pd
 from typing import Literal
 
 from .. import B_os
+# from byzh.core import B_os
 
 class B_Record2d:
-    def __init__(self, csv_path, mode:Literal["w", "a"]="w"):
+    def __init__(self, csv_path, mode:Literal["w", "a", "r"]="r"):
         self.csv_path = csv_path
+        self.mode = mode
 
         B_os.makedirs(csv_path)
         self.data = pd.DataFrame()
@@ -14,19 +16,45 @@ class B_Record2d:
             self.__read()
         elif mode == "a":
             self.__read()
+        elif mode == "r":
+            self.__read()
 
     # 支持 recorder[row, col] 访问
     def __getitem__(self, key):
-        row, col = key
-        row, col = str(row), str(col)
-        return self.data.loc[row, col]
+        if isinstance(key, tuple) and len(key) == 2: # 读值操作
+            row, col = key
+            row, col = str(row), str(col)
+            return self.data.loc[row, col]
+        if isinstance(key, str): # 读行操作
+            row = str(key)
+            return self.data.loc[row, :]
 
     # 支持 recorder[row, col] = value
     def __setitem__(self, key, value):
-        row, col = key
-        row, col, value = str(row), str(col), str(value)
-        self.data.loc[row, col] = value
+        assert self.mode!="r", "不允许读模式下赋值"
+
+        if isinstance(key, tuple) and len(key) == 2: # 赋值操作
+            row, col = key
+            row, col, value = str(row), str(col), str(value)
+            self.data.loc[row, col] = value
+            self.__save()
+        if isinstance(key, str): # 赋值行操作
+            assert type(value) == pd.Series, "仅支持 Series 赋值"
+            key = str(key)
+            self.data.loc[key, :] = value.copy()
+
+    # 复制一行
+    def copy_row(self, new_row, origin_row):
+        new_row, origin_row = str(new_row), str(origin_row)
+
+        if origin_row not in self.data.index:
+            raise KeyError(f"源行 {origin_row} 不存在")
+
+        # 深拷贝一份行数据
+        self.data.loc[new_row] = self.data.loc[origin_row].copy()
+
         self.__save()
+        return self
 
     # 转置
     def T(self):
@@ -51,10 +79,10 @@ class B_Record2d:
         return float(self[row, col])
 
     def get_bool(self, row, col) -> bool:
-        result = self[row, col]
-        if result in ("True", "true", "1"):
+        result:str = self[row, col]
+        if result.lower() in ("true", "1"):
             return True
-        elif result in ("False", "false", "0"):
+        elif result.lower() in ("false", "0"):
             return False
         else:
             raise ValueError(f"无法转换为布尔值 -> {result}")
@@ -85,6 +113,8 @@ if __name__ == '__main__':
     recorder["qwq", "OvO"] = 30
     recorder["qwq", "TwT"] = 40
 
+    recorder['aaa'] = recorder['qwwq']
+
     print("当前内容：")
     print(recorder)
 

{byzh_core-0.0.9.2 → byzh_core-0.0.9.4}/byzh_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: byzh-core
-Version: 0.0.9.2
+Version: 0.0.9.4
 Summary: byzh-core是byzh系列的核心库，包含了一些常用的工具函数和类。
 Author: byzh_rc
 License: MIT