如果一个元素等于零，我如何让我的输出排除它？[重复]

Question

所以我在Stack Overflow用户的帮助下创建了这段代码。

def get_name(string):
    return string.replace("+", "").replace("-", "")

def gnames(input_list: list):
    output = {}
    for entry in input_list:
        if '->' in entry:
            names = entry.split('->')
            output[names[1]] = output[names[0]]
            output[names[0]] = 0
        else:
            name = get_name(entry)
            if name not in output:
                output[name] = 0
            if "++" in entry:
                output[name] += 1
            if "--" in entry:
                output[name] -= 1
    return output
print(gnames(["Jim--", "John--", "Jordan--", "Jim++", "John--", "Jeff--", "June++", "June->Jim"]))

这就回来了

{'Jim': 1, 'John': -2, 'Jordan': -1, 'Jeff': -1, 'June': 0}

现在这是正确的，但是我希望gnames（）只返回非零值负数或正数都可以

在我的例子中，有'June'=0

我想让gnames（）的输出排除'June'=0或者如果任何其他人有一个0。。。我想要gnames（）排除它。。。

因此，在这种情况下，我的输出应该返回

{'Jim': 1, 'John': -2, 'Jordan': -1, 'Jeff': -1}

我怎么能这么做？？

Answer 1

如果要避免在中使用显式的，请先检查字典中是否存在该键，否则请将默认值设置为0，您可以使用集合。defaultdict是一种dict子类型，如果键不在dict中，它会自动设置默认值。

因此，您的代码可以简化为default法规（int），然后用一个符号到delta（更改）映射进行轻微改进，最后，如果添加的结果是，您可以选择从法规中删除键0值；这可以用来避免在返回语句中完全需要判断理解。

现在添加了所有更改后，端到端代码如下所示：

from collections import defaultdict


def get_name(string):
    return string.replace("+", "").replace("-", "")


def gnames(input_list: list):
    output = defaultdict(int)
    sign_to_delta = {'++': 1,
                     '--': -1}

    for entry in input_list:
        if '->' in entry:
            from_, to = entry.split('->')
            # Use `pop` to retrieve and also delete the key from dict
            # instead of assigning the value as 0, which is already the
            # default behavior if the key is not present.
            output[to] = output.pop(from_)
        else:
            name = get_name(entry)
            for sign in sign_to_delta:
                if sign in entry:
                    result = output[name] + sign_to_delta[sign]
                    if result:
                        output[name] = result
                    else:
                        del output[name]

    return output


print(gnames(["Jim--", "John--", "Jordan--", "Jim++", "John--", "Jeff--", "June++", "June->Jim"]))
print(gnames(["Jean++", "Jean--", "Jean--", "Jean++"]))

输出：

defaultdict(<class 'int'>, {'John': -2, 'Jordan': -1, 'Jeff': -1, 'Jim': 1})
defaultdict(<class 'int'>, {})

Answer 2

此外，使用字典理解，您可以更改您的报税表，如下所示：

return {x:output[x] for x in output if output[x] != 0}

这是做什么的描述：

# for each element in output
#    if element['name'] != 0 then keep it

完整代码：

def get_name(string):
    return string.replace("+", "").replace("-", "")

def gnames(input_list: list):
    output = {}
    for entry in input_list:
        if '->' in entry:
            names = entry.split('->')
            output[names[1]] = output[names[0]]
            output[names[0]] = 0
        else:
            name = get_name(entry)
            if name not in output:
                output[name] = 0
            if "++" in entry:
                output[name] += 1
            if "--" in entry:
                output[name] -= 1
    return {x:output[x] for x in output if output[x] != 0}


print(gnames(["Jim--", "John--", "Jordan--", "Jim++", "John--", "Jeff--", "June++", "June->Jim"]))

# Sample output
# {'Jim': 1, 'John': -2, 'Jordan': -1, 'Jeff': -1}

Answer 3

字典理解使这变得非常简单。如果我们从gnames为{'Jim'： 1，'John'：-2，'Jordan'：-1，'Jeff'：-1，'June'： 0}开始，我们可以编写一个字典理解来过滤掉零值。

>>> gnames = {'Jim': 1, 'John': -2, 'Jordan': -1, 'Jeff': -1, 'June': 0}
>>> {k: v for k, v in gnames.items() if v != 0}
{'Jim': 1, 'John': -2, 'Jordan': -1, 'Jeff': -1}
>>>