如何根據閾值化簡一個pandas dataframe

Question

這是我的 dataframe

      A       B        C        D
1     0    0.41     0.35     0.61
2     0    0.41     0.35        0
3     0    0.21        0        0
4  0.11     0.4     0.53        0

我只想顯示包含值大於 0.5 的列或行，就像這樣

       C        D
1    0.35     0.61
4    0.53        0

我應該怎么做

Answer 1

使用DataFrame.gt測試更大的值，使用DataFrame.any測試是否匹配至少一個值並在DataFrame.loc中過濾：

m = df.gt(0.5)
df1 = df.loc[m.any(axis=1), m.any()]
print (df1)
     C     D
1  0.35  0.61
4  0.53  0.00

Answer 2

您可以使用雙boolean 索引。 使用gt計算 boolean 掩碼。 然后檢查每個軸上是否有any值是 True 並將其用於使用loc的選擇

m = df.gt(0.5)
df.loc[m.any(1), m.any(0)]

output：

      C     D
1  0.35  0.61
4  0.53  0.00

中間體

m ：

       A      B      C      D
1  False  False  False   True
2  False  False  False  False
3  False  False  False  False
4  False  False   True  False

m.any(1) ：

1     True
2    False
3    False
4     True
dtype: bool

m.any(0) ：

A    False
B    False
C     True
D     True
dtype: bool