python - 如何将同一行中的列作为熊猫中的 .loc 进行子集化

Question

这里的效率问题。有时我想运行 a.loc并在这样做之后查看某些列。我一直这样做的方式如下

p = df[(df.x==1)]
p(['x','y'])

这将返回值为1from column的单元格，x然后对数据框进行子集化以仅查看列x和y. 有没有一种方法可以在没有变量赋值的情况下在一行中做到这一点？我尝试过的事情包括

df[(df.x==1)].columns(['x','y'])

df[(df.x==1),['x','y'])

我经常这样做，如果我能把它变成一行代码，我会很高兴。谢谢！

标签： pythonpandas

Answer 1

不要链接你的索引。而不是 l，使用loc

df.loc[df.x==1, ['x','y']]

一般语法是

df.loc[index_slice, column_slice]