通过索引位置和比较运算符过滤df

Question

我试图通过对最后两列的浮点值小于1.0的值less进行子集设置来过滤df。

b = np.array([[10.0,2.0,3.0,7.0], [10.0,99.0,20.0,63.0], [0.5,99.0,20.0,63.0] ])
df = pd.DataFrame(b.T) 
cols = df.iloc[:,-2:].columns.values
df = df[df[cols] < 1.0]

输出：

0   1    2
0 NaN NaN  0.5
1 NaN NaN  NaN
2 NaN NaN  NaN
3 NaN NaN  NaN

预期：

0    1    2
0 10.0 10.0  0.5

Answer 1

你的意思是：

df[df[cols].le(1).any(axis=1)]