使用列從數據框中選擇多行

Question

因此，我有一個包含很多行的數據框，我試圖使用一列數據只選擇其中一些行的一列。 是否有命令要求具有特定值的列並返回包含該列的行？ 哦，我只想選擇返回的行中的一列。

Answer 1

您正在尋找df.loc

例如：

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

返回col1 == 3的所有行

要么：

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

返回與上面的數據幀類似的數據幀，但它只有col1和col2

Answer 2

該文檔對此提供了快速指南。

要選擇滿足特定條件的行，可以使用方括號。

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

這將返回一個子集數據框，因此您可以像通常那樣選擇列。

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