imaris导出数据的提取和整理

1
from glob import glob
2
import pandas as pd
3

4

5
def collect_surface_basic_attributes(name):
6
    '''Get Basic Attributes from Imaris Surface Statistics Detailed Results
7
    name: the string input when Export All Statistics to File in Imaris, resulting a `{name}_Statistics` folder
8
    The location (x, y, z) of mass centroid, the mean intensity and the volume of the surface object would be collected.
9
    '''
10
    # 整理数据表格文件路径
11
    fps1 = glob(f"{name}_Statistics/{name}_Center_of_Image_Mass_*.csv")
12
    fps2 = glob(f"{name}_Statistics/{name}_Intensity_Mean_*.csv")
13
    fps3 = glob(f"{name}_Statistics/{name}_Volume*.csv")
14
    fps = {"locs": fps1[0],
15
        "intensity": fps2[0],
16
        "volume": fps3[0]}
17
    # 读取数据表格, 指定第二行作为表头header开始读表
18
    a = pd.read_csv(fps['locs'], header=2)
19
    b = pd.read_csv(fps['intensity'], header=2)
20
    c = pd.read_csv(fps['volume'], header=2)
21

22

23
    # 根据唯一的ID来汇总所有数据
24
    box = {}
25
    for idx, row in a.iterrows():
26
        if row.ID not in box:
27
            box[row.ID] = {}
28
        box[row.ID]['x'] = row['Center of Image Mass X']
29
        box[row.ID]['y'] = row['Center of Image Mass Y']
30
        box[row.ID]['z'] = row['Center of Image Mass Z']
31
        box[row.ID]['unit'] = row['Unit']
32
    for idx, row in b.iterrows():
33
        box[row.ID]['mean'] = row['Intensity Mean']
34
    for idx, row in c.iterrows():
35
        box[row.ID]['volume'] = row['Volume']
36

37

38
    # 转换为dataframe对象方便保存为表格文件
39
    data = pd.DataFrame(box).T
40
    data.index.name = 'ID'
41
    data.to_csv(f"{name}_collection.csv")
42
    return data