【问题标题】:Split list of tuples into seprate list based on a condition根据条件将元组列表拆分为单独的列表
【发布时间】: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


【解决方案1】:

我在这里发布@Mustafa Aydin 的解决方案,以便其他用户可以尽快找到解决方案。

inds, = np.nonzero(np.array(a)[:, 0] == 1)
[sub.tolist() for sub in np.split(a, inds) if sub.size]

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2021-05-21
    • 1970-01-01
    • 2020-11-08
    • 1970-01-01
    • 2010-10-31
    • 1970-01-01
    相关资源
    最近更新 更多