python-3.x - 从熊猫数据框中提取数据

Question

我想根据列值从更大的数据框创建数据框列表。例如，该列"ID"可以重复1,2,3,1,2,3,4,5,1,2。

我想通过提取行来创建数据框列表，直到 ID 再次重复回到 1。在这种情况下，列表应该有 3 个数据，ID 为：1,2,3，1,2,3,4,5然后1,2。

这可以在不使用 for 循环的情况下完成吗？

标签： python-3.xpandaspandas-groupby

Answer 1

不需要循环。

>>> list(zip(*df.groupby(df.ID.diff().ne(1).cumsum())))[1]