熊貓中2個指標列的邏輯或

Question

我有一個有兩個指標變量列的數據框。我想創建一個第三列，它是兩個現有列的邏輯OR。我能夠，像這樣：

df  = DataFrame ({"a" : [1,0,0] , "b" : [0,0,1] }) 
df["c"] = (df["a"] | df["b"]).apply(lambda x : 1 if x else 0) 

通過(df["a"] | df["b"])生產的系列bool類型。這令我感到驚訝，因爲|是一個按位運算符，所以我預計該系列的類型爲int。因此，我必須執行apply(lambda ...)才能獲得所需的int列。

我是否正確地做了這件事，還是有一個更簡潔/大熊貓首選/ pythonic的方式？

Answer 1

使用astype：

df["c"] = (df["a"] | df["b"]).astype(int) 

的|是不是真正的位運算符。它會調用方法__or__。 這可以根據需要實現。

所以這個：

df['a'] | df['b'] 

實際上是調用此：

df['a'].__or__(df['b'])