python有条件地访问列表列表

Question

我想就有关访问存储在列表列表中的数据的问题提出一些建议。

我的列表列表如下所示:

my_list = [['RN1', 200, 0.55], ['RN2', 0.8, 0.44], ['RN3', 0.2, 0.01]]

列表根据内部列表的第三个元素排序。

我想做的是:

1. 检查第一个内部列表的第三个元素是否大于或等于 0.9；如果是，我想创建一个包含第一个内部列表的前 2 个元素的列表

   my_list[0][2] >= 0.9
   

   如果为 true 将给出:

   new_list = [['RN1', 200]]
   

2. 如果没有，我想将第二个内部列表的相应元素添加到 my_list[0][2] 元素:

   my_list[0][2] + my_list[1][2] 并检查此值是否大于 0.9。

如果是，我想停止将第二个内部列表的前 2 个元素添加到我的 new_list 中的循环:

new_list = [['RN1', 200], ['RN2', 0.8]]

如果不继续添加元素，直到达到 0.9 的阈值。

当然这只是一个例子。我的真实列表可能要大得多，而且这个过程应该是迭代的。

我尝试了不同的可能性，但所有方法均无效。例如我做了:

sum_percentage = 0.0

new_list =[]

    for i in old_list:

        if sum_percentage <= 0.9:

            sum_percentage = sum_percentage + i[2]

        new_list.append([i[0], i[1]])

还有其他变体……现在我卡住了。

有什么建议吗？

非常感谢

Answer 1

这将为您完成工作:

sum_percentage = 0.0
new_list = []
for i in old_list:
    new_list.append(i[:2])
    sum_percentage += i[2]
    if sum_percentage >= 0.9:
        break

我认为这是不言自明的，但如果您对任何事情感到困惑，请告诉我，我会扩展这个答案。