0

我需要遍历一个列表，并删除所有不满足某些条件的元素。具体来说，每当我有两个连续的元素 A 和 B 时，其中 A 为负数，B 为正数（所有元素都非零），并且这些元素的比率小于某个给定比率，它们应该被删除。所有其他元素都应该保留。

起初，我在列表中循环，然后直接删除。但是，由于我正在循环遍历列表的长度，并且循环实际上正在改变这个长度，所以我遇到了索引错误。所以我想我会创建一个包含所有满足条件的元素的新列表，而不是删除。这是正确的方法吗？这是我的代码：

d_list = [1, -6, 7, -10, 11, -12, 13, -15, 17, -18, 19, -20, 23, -24, 29, -30, 31]
J = []
for i in range(0, len(d_list) - 1):
    if ((d_list[i] < 0) & (d_list[i+1] > 0)):
        if(math.fabs(d_list[i+1] / d_list[i]) >= 6/5):
            J.append(d_list[i])
            J.append(d_list[i+1])
    else:
        J.append(d_list[i])

我应该有输出：[1, -24, 29].

但是我得到：[1, 7, 11, 13, 17, 19, 23, -24, 29, 29]。

我非常困惑，并尝试了上面显示的代码的几种变体，但均未成功。

4

1 回答 1

1

干得好：

import math

d_list = [1, -6, 7, -10, 11, -12, 13, -15, 17, -18, 19, -20, 23, -24, 29, -30, 31]
J = []
skip = []

for i in range(0, len(d_list) - 1):
    if i in  skip:
        continue
    if d_list[i] < 0 < d_list[i+1]:
        if math.fabs(d_list[i+1] / d_list[i]) >= 6/5:
            J.append(d_list[i])
        else:
            skip.append(i+1)

    else:
        print(d_list[i])
        J.append(d_list[i])

print(J)

Output:[1, -24,-29]

问题是您要附加两个元素而不是一个，如果元素无效，您应该在下一次检查中跳过它，因为它已经不符合条件。例如我们有

A->B->C

如果 A->B 失败，您应该将它们都删除，并且您的代码省略 A，然后检查 B->C 是否有效，因此您将它们都添加，但您不应该添加它们。

如果您希望我可以详细说明，我希望这很清楚。

编辑：您也可以使用 bool 值来跳过列表，如下所示：

import math

d_list = [1, -6, 7, -10, 11, -12, 13, -15, 17, -18, 19, -20, 23, -24, 29, -30, 31]
J = []
skip = False
for i in range(0, len(d_list) - 1):
    if skip:
        skip = False
        continue
    if d_list[i] < 0 < d_list[i+1]:
        if math.fabs(d_list[i+1] / d_list[i]) >= 6/5:
            J.append(d_list[i])
        else:
            skip =True

    else:
        print(d_list[i])
        J.append(d_list[i])

print(J)

Output:[1, -24,-29]

但是，第一种方法保留了为什么删除元素的信息。

于 2019-05-13T22:24:51.373 回答