如何按iloc

Question

我有一个包含2列的数据框。第二列是仅有的几个值之一。我想创建一个返回数据框的方法，其中只包含该列具有特定值的行。

我使用了这段代码：

def filterOnName(df1):
    d1columns = df1.columns
    return df1[df1[d1columns[1]] == "Jimmy"]

看起来很复杂并不是吗？我想有一个叫做iloc的熊猫方法应该清理这个升技但是我在实现它时遇到了麻烦。你能解释一下我做错了吗？

def filterOnName(df1):
    return df1[df1.iloc[1] == "Jimmy"]

感谢您的帮助！

Answer 1

.iloc的第一个参数是行。要获得第二列，您需要：

df.iloc[:, 1]

其中:表示“所有行”。