如何通过多列获取 Pandas DataFrame 中每组的第一行？

通过多列获取 Pandas DataFrame 中每个组的第一行

在 pandas DataFrame 中，通常需要检索第一个根据特定列对 DataFrame 进行分组后每组的行。使用 pandas 的内置方法可以有效地完成此任务。

要实现此目的，可以使用 .first() 方法获取每个组的第一个（非空）元素。 first() 的语法如下：

在您的示例中，您希望按“id”和“value”列对 DataFrame 进行分组，并获取每个组的第一行。您可以使用以下代码：

这将产生以下结果：

id	value
1	first
2	first
3	first
4	second
5	first
6	first
7	fourth

如果您希望将“id”作为结果 DataFrame 中的列，您可以使用reset_index()重置索引，如下所示：

此操作的输出将be:

id	value
1	first
2	first
3	first
4	second
5	first
6	first
7	fourth

另外，如果你想检索每组中的前n行，你可以使用.head()方法而不是first()。例如，要获取每组的前两行，您可以使用：

这将返回以下 DataFrame：

id	value
1	first
1	second
2	first
2	second
3	first
3	third
4	second
4	fifth
5	first
6	first
6	second
7	fourth
7	fifth

以上是如何通过多列获取 Pandas DataFrame 中每组的第一行？的详细内容。更多信息请关注PHP中文网其他相关文章！