first commit

Code/7-box-graph.py

@@ -0,0 +1,50 @@
+import openpyxl
+import os.path
+import matplotlib.pyplot as plt
+
+
+def draw_box_graph(region, data, groups, data_type, save_path):
+    plt.rcParams["font.sans-serif"] = ["SimHei"]  # 设置字体
+    plt.boxplot(data,
+                patch_artist=True,  # 是否填色
+                labels=groups)
+
+    plt.title(f"各个分组{region}{data_type}对比")
+    plt.xlabel("分组")
+    plt.ylabel(data_type)
+    plt.savefig(save_path, dpi=200)
+    plt.cla()
+    plt.close('all')
+
+
+def main():
+    data_types = ['corr-CBF', "ATT"]
+    # 请尝试添加'AnImage_AAL3'图谱,并修复错误
+    atlas_list = ['AnImage_WholeBrain', 'AnImage_BrainLobes']
+    output_dir = r'..\Data\box-graph'
+    os.makedirs(output_dir, exist_ok=True)
+
+    input_file = r'..\Data\csv-group.xlsx'
+    workbook = openpyxl.load_workbook(input_file, data_only=True)
+    groups = ['A', 'B', 'C']
+
+    for data_type in data_types:
+        for atlas in atlas_list:
+            value_dict = {}
+            for group in groups:
+                worksheet = workbook[f'{group}_{atlas}_{data_type}']
+                for column_num in range(2, worksheet.max_column + 1):
+                    data = []
+                    name = worksheet.cell(row=1, column=column_num).value
+                    for row in range(2, worksheet.max_row - 10):
+                        data.append(worksheet.cell(row=row, column=column_num).value)
+
+                    if name not in value_dict:
+                        value_dict[name] = []
+                    value_dict[name].append(data)
+
+            for region, data in value_dict.items():
+                draw_box_graph(region, data, groups, data_type, os.path.join(output_dir, f'{region}{data_type}.png'))
+
+
+main()