genods.py 文件源码

def loadCSV(csvfile):
        """ Load the results
        Retuns: ID list of ints, ROI (array) of strings
        """
        # get information about the slices first
        vcnt=5  # we expect 5 error measures
        values = {}
        with open(csvfile, 'rb') as csvfile:
                reader = csv.reader(csvfile, delimiter=',', quoting=csv.QUOTE_NONE)
                values = {}
                cnt = 0
                for row in reader:
                        if(reader.line_num == 1): 
                            apps=[]
                            for i in range(len(row)):
                                if row[i] != '': apps.append(row[i])
                            apps = apps[1:]
                        elif(reader.line_num == 2): 
                            labels=row[1:vcnt+1]
                        elif(reader.line_num > 2): 
                            d={}
                            for i in range(len(apps)):
                                d[apps[i]]=row[1+vcnt*i: 1+vcnt*(i+1)]
                            values[row[0]] = d
                        cnt +=1
                        #if cnt > 100: break
        return apps, labels, values