gấu trúc tương đương với np.where

Question

np.where có ngữ nghĩa của một véc tơ nếu/else (tương tự như phương pháp của Apache Spark when/otherwise DataFrame). Tôi biết rằng tôi có thể sử dụng np.where trên gấu trúc Series, nhưng pandas thường xác định API riêng của mình để sử dụng thay vì các hàm numpy thô, thường thuận tiện hơn với pd.Series/pd.DataFrame.

Chắc chắn đủ, tôi đã tìm thấy pandas.DataFrame.where. Tuy nhiên, ngay từ cái nhìn đầu tiên, nó có một ngữ nghĩa hoàn toàn khác. Tôi không thể tìm thấy một cách để viết lại ví dụ cơ bản nhất của np.where sử dụng gấu trúc where:

# df is pd.DataFrame 
# how to write this using df.where? 
df['C'] = np.where((df['A']<0) | (df['B']>0), df['A']+df['B'], df['A']/df['B']) 

Am tôi thiếu một cái gì đó rõ ràng? Hoặc là gấu trúc where dành cho trường hợp sử dụng hoàn toàn khác, mặc dù có cùng tên là np.where?

Answer 1

Hãy thử:

(df['A'] + df['B']).where((df['A'] < 0) | (df['B'] > 0), df['A']/df['B']) 

Sự khác biệt giữa numpywhere và DataFramewhere là các giá trị mặc định được cung cấp bởi các DataFrame rằng phương pháp where đang được kêu gọi (docs).

I.e.

np.where(m, A, B) 

là tương đương với

A.where(m, B) 

Nếu bạn muốn có một chữ ký cuộc gọi tương tự như sử dụng gấu trúc, bạn có thể tận dụng lợi thế của the way method calls work in Python:

pd.DataFrame.where(cond=(df['A'] < 0) | (df['B'] > 0), self=df['A'] + df['B'], other=df['A']/df['B']) 

hoặc không kwargs (Lưu ý: rằng thứ tự vị trí của đối số khác với số numpywhereargument order):

pd.DataFrame.where(df['A'] + df['B'], (df['A'] < 0) | (df['B'] > 0), df['A']/df['B'])