where() function 系列與 DataFrame 需要解釋

Question

df 有 A、B、C、D、E 列，假設“A”列是字符串，rest 是數字。

df["A"].where(df[B] > 100).dropna()返回列“A”，只要“B”的值 > 100

我的問題是df["A"] （它是原始df的視圖）沒有“B”列，那么“where”子句如何應用於B列。[where（）子句應用於df["A"]但不是整個 "df"]

df["A"]的類型是 Pandas 系列，即使如此，“B”列上的where()子句也有點令人困惑如何應用它。

Answer 1

很容易得到 dataframe 的所有列，而不僅僅是A 。

只需刪除["A"]部分：

df["A"].where(df["B"] > 100).dropna()

至

df.where(df["B"] > 100).dropna()

現在您可以執行以下操作：

>>> subset = df.where(df["B"] > 100).dropna()
>>> subset["B"]
...
>>> subset["A"]

---

注意：不是使用where + dropna ，而是以下更短的等效解決方案。

代替

df.where(df["B"] > 100).dropna()

只需使用

df[df["B"] > 100]