Python DataFrame - Select dataframe rows based on values in a column of same dataframe

Question

我正在努力解决与 dataframe 相关的问题。

columns = [desc[0] for desc in cursor.description]
data = cursor.fetchall()
df2 = pd.DataFrame(list(data), columns=columns)

df2如下：

| Col1     | Col2           |
| -------- | -------------- |
| 2145779  | 2              |
| 8059234  | 3              |
| 2145779  | 3              |
| 4265093  | 2              |
| 2145779  | 2              |
| 1728234  | 5              |

我想在 col1 中列出值 col2="3"

Answer 1

您可以使用 boolean 索引：

out = df2.loc[df2.Col2.eq(3), "Col1"].agg(list)
print(out)

印刷：

[8059234, 2145779]