python - 如何将条件应用于 pandas iloc

Question

我从 pandas DataFrame 中选择 2 - end 列，iloc 为

d=c.iloc[:,2:]

现在如何将条件应用于此选择？例如，如果 column1==1。

Answer 1

您可以使用 DataFrame.iloc如果需要按位置筛选第一列，: 表示这里选择所有行:

c[c.iloc[:, 0] == 1]

示例:

c = pd.DataFrame({'A':list('abcdef'),
                   'B':[4,5,4,5,5,4],
                   'C':[7,8,9,4,2,3],
                   'D':[1,3,5,7,1,0],
                   'E':[5,3,6,9,2,4],
                   'F':list('aaabbb')})

print (c)
   A  B  C  D  E  F
0  a  4  7  1  5  a
1  b  5  8  3  3  a
2  c  4  9  5  6  a
3  d  5  4  7  9  b
4  e  5  2  1  2  b
5  f  4  3  0  4  b

df = c[c.iloc[:, 3] == 1]
print (df)
   A  B  C  D  E  F
0  a  4  7  1  5  a
4  e  5  2  1  2  b