Python Добавить ключ к Dict условно

Question

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

headers = ["A_1","A_2","A_3","B_1","B_2","B_3"] 

в

cols = { 1 : {'A' : 0, 'B' : 3}, 2: {'A' : 1, 'B' : 5} ... } 

Мой код выглядит следующим образом:

cols = {} 
headers = ["A_1","A_2","A_3","B_1","B_2","B_3"] 

col_number = 0 
for header in headers: 
    run_number = header[-1:] 
    cols[ run_number ] = {} 

    if "A_" in header: 
     cols[ run_number ][ 'A' ] = col_number 
    if "B_" in header: 
     cols[ run_number ][ 'B' ] = col_number 

    col_number += 1 
print cols 

Этот вывод только последний "B" колонки:

{'1': {'B':3}, '2':{'B':5}... } 

ОДНАКО, если я попробую простой эксперимент t в командной оболочке, эта нотация, похоже, хорошо работает ...

cols = {} 
cols[1] = {} 
cols[1]['A'] = 1 
cols[1]['B'] = 2 
print cols 

>> {'1' : {'A':1,'B':2} } 

Почему?

EDIT: Просто нужен еще один набор глаз, я думаю. Проблема была эта линия в петле ...

cols[ run_number ] = {} 

Он переписывает COLS [run_number] в пустой Словаре каждый раз, когда колонна из предварительно существующих run_number достигается.

Answer 1

Ваш цикл перезаписи этой важной переменной и установить его в пустой Dict:

cols[ run_number ] = {} 

Это означает, что он получает через струнам «A_» для различных чисел, а затем, когда он на струнах «B_» , он перезаписывает все ключи run_number, которые уже созданы для строк «A_».

Вы должны попробовать что-то вроде следующего вместо:

if run_number not in cols: 
    cols[ run_number ] = {} 

Кроме того, вы можете попробовать использовать defaultdict

Answer 2

Вы должны перейти к следующей строке кодов:

if run_number not in cols: 
    cols[ run_number ] = {} 

I испытано, работает следующий код:

for header in headers: 
    run_number = header[-1:] 
    if run_number not in cols: 
     cols[ run_number ] = {} 
    if "A_" in header: 
     cols[ run_number ][ 'A' ] = col_number 
    if "B_" in header: 
     cols[ run_number ][ 'B' ] = col_number 
    col_number += 1 

Answer 3

Во-первых, вы можете использовать enumerate(), чтобы Python автоматически отслеживал подсчет. Во-вторых, как говорили другие, вы заменяете свои словарные ключи новым, а не добавляете в существующий словарь.

Нечто подобное должно работать для вашего случая:

cols = {} 
for col, header in enumerate(headers): 
    typ, _, key = header.partition('_') 
    if key not in cols: 
     cols[key] = {} 
    cols[key][typ] = col 

с вашими данными, cols является:

{'1': {'A': 0, 'B': 3}, '2': {'A': 1, 'B': 4}, '3': {'A': 2, 'B': 5}} 

Answer 4

1. Поскольку "run_number" переменная повторяется каждый раз, когда вы цикл и он повторяется, вы перезаписываете его любым пустым dict.

2. Ниже приведен способ достижения вашей цели.

   def answer(): 
       headers = ["A_1","A_2","A_3","B_1","B_2","B_3"] 
       cols = {} 
       cols[1] = {} 
       cols[2] = {} 
       cols[3] = {} 
       for header in headers: 
        cols[int(header[-1:])][str(header[:-1])] = headers.index(header) 
   
       return cols 
   
   print answer()