使用列从数据框中选择多个行

Question

所以，我有一个包含很多行的数据框，我试图使用一列数据只选择其中一些行的列。是否有一个命令要求具有特定值的列并返回包含该列的行？哦，我想只选择返回行的一列。

Answer 1

您正在寻找df.loc

例如：

df.loc[df['col1']==3]

返回col1 == 3

的所有行

或者：

df.loc[df['col1']==3, ['col2', 'col1']]

返回类似上面的数据框，但它只有col1和col2

Answer 2

documentation有一个快速指南。

要选择符合特定条件的行，可以使用括号。

df[(condtion)]
# ex.
df[df['A'] > 3]

返回子集化数据框，因此您可以像往常一样选择列。

df[df['A'] > 3][['A', 'B']]