如何在python中选择具有相同id的列表？

问题描述

我有一个如下所示的数据框：

id	地点	年龄	性别
13	1	3	1
13	2	4	1
13	3	3	2
13	4	4	2
14	1	3	1
14	2	4	1
14	3	3	2

我想选择 place、age 和 gender，其中 id 在 Python 中是唯一的。例如对于id=13，我想选择矩阵：

地点	年龄	性别
1	3	1
2	4	1
3	3	2
4	4	2

请注意，id 的长度不同。 感谢您的帮助。

解决方法

您只需执行 id 即可选择所有 df[df['id'] == 13] 为 13 的行。

如果您只想要剩余的列 place、age 和 gender，则：

df.loc[df['id'] == 13,['place','age','gender']]
# or
df[df['id'] == 13][['place','gender']]