python - 如何通过iloc过滤

Question

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

我用这个代码工作:

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

看起来很复杂不是吗？我想有一个名为 iloc 的 pandas 方法应该可以解决这个问题，但我在实现它时遇到了麻烦。你能解释一下我做错了什么吗？

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

感谢您的帮助!

Answer 1

.iloc 的第一个参数用于行。要获得第二列，您需要:

df.iloc[:, 1]

其中 : 表示“所有行”。