时间戳

首页 > 解决方案 > 'int' 对象不可迭代”当使用 itertools 并将函数应用于每一行时

python - 'int' 对象不可迭代”当使用 itertools 并将函数应用于每一行时

问题描述

我有以下数据集：

index    REWARD  
(1,1,1)   0
(1,2,3)   0
(1,1,3)   0

如果索引有一对数字，我想设置 REWARD = 2。所以输出应该看起来像

index    REWARD  
(1,1,1)   0
(1,2,3)   0
(1,1,3)   2

当我使用此代码时

  def set_reward(final):
        for i in final['index']:
            tempCount=[]
            for item,count in collections.Counter((i)).items():
                tempCount.append(count)
            if tempCount==[2, 1] or tempCount==[1, 2]:
                final['REWARD']=2
            return final['REWARD']
    final['REWARD']=final.apply(set_reward,axis=1)

它说'int'对象是不可迭代的”

有什么办法可以解决吗？

标签： pythonpandas

解决方案

您可以在没有显式 for 循环和条件逻辑的情况下获得所需的结果。尝试这样的事情：

# Example data
df = pd.DataFrame({'index':  [(1, 1, 1), (1, 2, 3), (1, 1, 3)], 
                   'REWARD': [0, 0, 2]})


# Select any row whose index contains at least one pair of values
mask = df['index'].apply(lambda x: 2 in Counter(x).values())

df.loc[mask, 'REWARD'] = 2

df
       index  REWARD
0  (1, 1, 1)       0
1  (1, 2, 3)       0
2  (1, 1, 3)       2

推荐阅读