jak połączyć dwie kolumny z pandami if/else w Pythonie?

Question

Jestem bardzo nowy w Pandach (to znaczy w mniej niż 2 dni). Jednak nie mogę znaleźć odpowiedniej składni do łączenia dwóch kolumn z warunkiem if/else.

Właściwie, wymyśliłem jeden sposób, aby to zrobić za pomocą "zip". To jest to, co chcę osiągnąć, ale wydaje się, że w pandach może być skuteczniejszy sposób.

Przez wzgląd na kompletność, I obejmują pewne wstępne przetwarzanie zrobić, aby było jasne:

records_data = pd.read_csv(open('records.csv')) 

## pull out a year from column using a regex 
source_years = records_data['source'].map(extract_year_from_source) 

## this is what I want to do more efficiently (if its possible) 
records_data['year'] = [s if s else y for (s,y) in zip(source_years, records_data['year'])] 

Answer 1

W pand> = 0.10.0 spróbować

df['year'] = df['year'].where(source_years!=0,df['year']) 

i zobacz:

http://pandas.pydata.org/pandas-docs/stable/indexing.html#the-where-method-and-masking

Jak zauważył w komentarzach, to używa np.where pod maską - różnica jest taka, że ​​pandy dostosowuje szereg z wyjściem (tak na przykład można zrobić tylko częściową aktualizację)

Answer 2

Może spróbuj np.where:

import numpy as np 
df['year'] = np.where(source_years,source_years,df['year'])