Заполнение dict из цикла for в Python

Question

У меня есть папка, содержащая подпапки (SEQ1, SEQ2, SEQ3 и т. Д.), Каждая из которых содержит 40 изображений.

Я хочу выполнить цикл над каждым изображением, выполнить операцию, которая дает значение, затем получить среднее значение из 40 значений в каждой подпапке и поместить это значение в dict как «значение» с названием подпапка - это «ключ».

Вот что я пробовал:

dic = {} 

for (dirname, dirnames, filenames) in os.walk('C:\Users\....'): 

for keys in dirnames: 
rec_3 = [] 


    for subdirname in filenames: 

     ds = dicom.read_file(os.path.join(dirname, subdirname)) #reading in DICOM images 
     lx, ly = ds.pixel_array.shape # beginning of operation 
     X, Y = np.ogrid[0:lx, 0:ly] 
     mask = (X - lx/2)**2 + (Y - ly/2)**2 < lx*ly/8 
     st_dev = np.std(ds.pixel_array[mask]) # end of operation 
     rec_3.append(st_dev) 
    rec_3_avg = np.mean(rec_3) 
    dic.update({keys:rec_3_avg}) 

print dic 

я получаю следующий результат "{ 'SEQ3': нан 'SEQ2': нан 'SEQ1': нан 'Seq4': нан}"

Мой идеальный результат имеет среднее значение 40 значений в каждой подпапке как значение здесь вместо «nan».

Помогите оценить очень.

Answer 1

Я хотел бы сделать словарь этих значений в подкаталоге первого и вычислить средства впоследствии, только личные предпочтения и легче читать.

dic_std_dev = {} 
base_path = '.' 
for root, dirs, files in os.walk(base_path): 
    #print dirs 
    for f in files: 
     if root != base_path: #If this is a subdirectory 
      ds = dicom.read_file(os.path.join(root, f)) #reading in DICOM images 
      lx, ly = ds.pixel_array.shape # beginning of operation 
      X, Y = np.ogrid[0:lx, 0:ly] 
      mask = (X - lx/2)**2 + (Y - ly/2)**2 < lx*ly/8 
      array = ds.pixel_array[mask] 
      if array: 
       st_dev = np.std(array) # end of operation 
       if root in dic_std_dev.keys(): 
        dic_std_dev[root].append(st_dev) 
       else: 
        dic_std_dev[root] = [st_dev] 

dic = {} 
for key in dic_std_dev.keys(): 
    if len(dic_std_dev[key]) > 0: 
     dic[key] = np.mean(dic_std_dev[key]) 

print dic 

Answer 2

Вы можете попробовать эту версию:

dic = {} 

for root, dirs, files in os.walk('C:\Users\....'): 
    for dirname in dirs: 
     if dirname.startswith('.'): 
      dirs.remove(dirname) 
    rec_3 = []   
    for filename in files:  
     ds = dicom.read_file(os.path.join(root, filename)) #reading in DICOM images 
     lx, ly = ds.pixel_array.shape # beginning of operation 
     X, Y = np.ogrid[0:lx, 0:ly] 
     mask = (X - lx/2)**2 + (Y - ly/2)**2 < lx*ly/8 
     array = ds.pixel_array[mask] 
     if array.all(): 
      st_dev = np.std(array) # end of operation 
      rec_3.append(st_dev) 
    rec_3_avg = np.mean(rec_3) 
    dic.update({os.path.basename(root):rec_3_avg}) 

print dic