如何制作将列表作为输入并返回总和的函数？

Question

我想创建一个函数，它将列表作为输入并返回列表元素之间的整数、实数和复数之和。 作为列表元素，数据类型计算和时排除整数、实数和复数以外的对象。

所以我就这样写了代码

def number_sum(lst):
    total = 0
    for e in lst:
        if type(e) == int or type(e) == float or type(e) == complex:
            total += e
    return total

x1 = [1, 3, 5, 7, 9]                     # 25
x2 = ["abc", [3, 5], 3.5, 2+3j, 4.5, 10] # 20 + 3j
x3 = []                                  # 0

print(number_sum(x1))
print(number_sum(x2))
print(number_sum(x3))

但我希望我可以在输入中获取 x1、x2、x3 列表。我该如何解决？

Answer 1

使用numbers.Number 检查值是否为数字。来自文档：

数字层次结构的根。如果你只是想检查一个 参数 x 是一个数字，不管是什么类型，使用 isinstance(x, 号）。

代码

import numbers


def number_sum(lst):
    def get_numbers(l):
        """Generator that returns a flatten view a list"""
        for e in l:
            if isinstance(e, numbers.Number):
                yield e
            elif isinstance(e, (tuple, list)):
                yield from get_numbers(e)

    return sum(get_numbers(lst))


x1 = [1, 3, 5, 7, 9]
x2 = ["abc", [3, 5], 3.5, 2 + 3j, 4.5, 10]

print(number_sum(x1))
print(number_sum(x2))

输出

25
(28+3j)

请注意，函数get_numbers 是一个生成器，它使用yield 和yield from 来展平和过滤嵌套列表。然后将get_numbers 的结果传递给内置函数sum 以实际求和。

Answer 2

您可以使用x = input("Enter the List: ")，以便用户可以在一行中输入列表。
导入 ast
如果您希望用户输入列表为 [1,2,3]，请使用
@ 987654323@
x = input()
x = ast.literal_eval(x)。
这会将字符串变成一个列表。
最后您的代码将如下所示：

import ast

x = input("Enter the List: ")
x = ast.literal_eval(x)

然后，执行您想要执行的必要功能
感谢@Matthis 和@S3DEV 提出改进建议