Создание переменной с использованием критериев из нескольких столбцов (PANDAS)

Question

У меня есть столбец с именем HousSize и столбец Independent. Я хочу, чтобы создать переменную, которая равна 1, если HouseholdSize равно 1 и Independent не равен 1. Мой текущий код выглядит следующим образом:

 df["HH1flag"] = df.where((df.HouseholdSize ==1) & (df.Independent != 1).notnull().astype(int)) 
      HH1flag = df[df.HH1flag == 1] 
      pd.DataFrame(HH1flag, columns=["LocIdent","HouseholdSize","Independent"]).to_csv(Targcsv, mode='ab') 

Я получаю следующее сообщение об ошибке: AttributeError: «поплавок» объект имеет no attribute 'all'

Редактировать: С некоторой помощью от EdChum я добавляю немного больше информации по этому вопросу. Упрощенная версия моего пересмотренного кода выглядит следующим образом:

import pandas as pd 
from pandas import * 
import csv 

FileName='HouseholdSizeTest.xlsx' 
data=pd.read_excel(FileName,'Sheet1') 
df=pd.DataFrame(data) 
Targcsv=('target.csv') 

HH1Text= [['Table C.1 HouseholdSize Check 1']] 
with open(Targcsv,'ab') as f: 
    titles=csv.writer(f) 
    titles.writerow([]) 
    titles.writerows(HH1Text) 
    titles.writerow([]) 

df["HH1flag"] = df[(df.HouseholdSize ==1) & (df.Independent != 1)].notnull().astype(int) 
HH1flag = df[df.HH1flag == 1] 
pd.DataFrame(HH1flag, columns=["LocIdent","HouseholdSize","Independent"]).to_csv(Targcsv, mode='ab') 

я получаю новую ошибку, для той же линии. Ошибка ValueError: неправильное количество элементов, переданных 3, размещение подразумевает 1. Я изо всех сил пытаюсь правильно вставить данные в это редактирование, но представьте три случая. первый с Independent = 1 и HousSize = 1, второй с Independent = 0 и HousSize = 3, а третий с Independent = 47 и HousSize = 1. Для третьего случая, HH1flag должен быть равен 1.

Answer 1

Если вы хотите столбец индикатора, то вы можете использовать np.where:

df["HH1flag"] = np.where((df.HouseholdSize ==1) & (df.Independent != 1), 1, 0) 

Так что это использует переданное в состоянии и, если истинные возвращает 1, если ложь он возвращает 0.

Причина, по которой мое первое предложение не срабатывало, было потому, что это возвращает серию с несколькими значениями столбцов, которые не то, что вы хотели.

Когда вы сделали df.where это не нравится пытается бросить свой ряд и поднял ставку AttributeError:

AttributeError: 'float' object has no attribute 'all'

Я не знаю, почему он это делает, нет ничего особенного в ваших данных, как я смог воспроизвести это с помощью простого df, состоящего только из ints.