làm thế nào để kết hợp hai cột với một if/else trong pandas python?

Question

Tôi rất mới đối với Pandas (tức là, dưới 2 ngày). Tuy nhiên, tôi dường như không thể tìm ra cú pháp đúng để kết hợp hai cột với điều kiện if/else.

Thực ra, tôi đã tìm ra một cách để thực hiện việc đó bằng cách sử dụng 'zip'. Đây là những gì tôi muốn thực hiện, nhưng có vẻ như có một cách hiệu quả hơn để làm điều này trong gấu trúc.

Đối với đầy đủ sake, tôi bao gồm một số tiền xử lý tôi làm gì để làm cho mọi việc rõ ràng:

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

Trong gấu trúc> = 0.10.0 thử

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

và xem:

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

Như đã đề cập trong các ý kiến, điều này không sử dụng np.where dưới mui xe - sự khác biệt là gấu trúc gắn loạt với đầu ra (ví dụ như vậy bạn chỉ có thể làm một bản cập nhật phần)

Answer 2

Có lẽ thử np.where:

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