简体   繁体   中英

Pandas: result column by aggregating entire row

I have a pandas dataframe containing tuples of booleans (real value, predicted value) and want to create new columns containing the amount of true/false positives/negatives. 熊猫数据框 I know i could loop through the indices and set the column value for that index after looping through the entire row, but i believe that's a pandas anti-pattern. Is there a cleaner and more efficient way to do this?

This seems to work fine:

def count_false_positives(row):
  count = 0
  for el in df.columns:
    if(row[el][0] and not row[el][1]): 
      count+=1
  return count

df.false_positives = df.apply(lambda row: count_false_positives(row), axis=1)

Another alternative would be to check the whole dataframe for (True,False) values and sum the amount of matches along the columns axis ( sum(axis=1) ).

df['false_positives'] = df.apply(lambda x: x==(True,False)).sum(axis=1)

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM