根据另一列提取列值

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'}