значения столбцов экстракта на основе другого столбца

Question

У меня есть две колонки, такие как это:

ветер люфта

да нет

нет да

нет нет

да

Если значение в столбце wind равно yes, я хочу извлечь соответствующие значения в столбце play, независимо от того, является ли значение в этом столбце yes или no.

Я написал этот цикл for, чтобы сделать это, но я не уверен, что это правильный способ сделать такую ​​операцию? tar_attr_labels - столбец play.

for val in labelFreq.keys():   # dict contains yes and no 

    dataSubset = [entry for entry in tar_attr_labels if val] 
    ..... 

Answer 1

Я считаю, что ваша labelFreq переменным имеет словарь с вашим wind столбца как ключ и play столбца в качестве значения.

Если это правда, то, что вы хотите сделать:

data_subset = {wind: play for wind, play in labelFreq.items() if wind == 'yes'}