【发布时间】:2021-07-24 10:47:23
【问题描述】:
我有一个这样的元组列表:
a = [(1, 0.0), (1, 1.0), (1, 0.0), (2, 0.36034115138592754), (1, 1.0),
(1, 0.0), (2, 0.0), (3, 0.9999999999999991), (4, 0.0), (5, 0.0),
(6, -1.0000000000000002), (7, 0.9999999999999999), (8, 1.0000000000000002), (9, 1.0),
(10, 0.0), (1, 1.0), (1, 0.0)]
我想每次第一个元素为 1 时拆分它。 我已经尝试过使用 groupby,但我每次都需要有一个不同的元组列表,因为以后我必须绘制它并且每个元组都是一个特定的点。 我的预期输出是:
[(1, 0.0)]
[(1, 1.0)]
[(1, 0.0), (2, 0.36034115138592754)]
[(1, 1.0)]
[(1, 0.0), (2, 0.0), (3, 0.9999999999999991), (4, 0.0), (5, 0.0),
(6, -1.0000000000000002), (7, 0.9999999999999999), (8, 1.0000000000000002), (9, 1.0), (10, 0.0)]
[(1, 1.0)]
[(1, 0.0)]
【问题讨论】:
-
请用您的预期输出更新您的问题。
-
完成了,我以为我已经写了,抱歉
-
@Ibrahim 我认为最后两个元组
(1, 1.0), (1, 0.0)也应该包含在预期输出中。 -
是的,他们应该是,这是为了给出我想要实现的目标。我还添加了最后两个
-
[sub.tolist() for sub in np.split(a, inds) if sub.size]其中inds, = np.nonzero(np.array(a)[:, 0] == 1)。
标签: python pandas list split tuples