[已解决]关于python数组计算问题

高义校长

有下面3个列表：

a=[1, 2, 3, 4, 5]
b=[1, 2, 3]
c=[1, 2, 3, 4, 5, [1, 2, 3]]

请问用代码如何写才能达到下面的结果。

（1）列表c所有元素求和，得到一个整数21。
（2）列表a和b对应元素相加，得到[2,4,6,4,5]这样一个列表

最佳答案

月排行榜 / 总排行榜

永恒的蓝色梦想

2020-4-2 16:36:46

高义校长 发表于 2020-4-2 16:31
不好意思，发太快，就是想问如何实现。

问题1

def func1(array):
        sum=0
        for i in array:
                if type(i) is list:
                        sum+=func1(i)
                else:
                        sum+=i
        return sum

跳转到最佳答案楼层

永恒的蓝色梦想

问题呢？？？

高义校长

永恒的蓝色梦想 发表于 2020-4-2 16:30
问题呢？？？

不好意思，发太快，就是想问如何实现。

永恒的蓝色梦想

高义校长 发表于 2020-4-2 16:31
不好意思，发太快，就是想问如何实现。

问题1

def func1(array):
        sum=0
        for i in array:
                if type(i) is list:
                        sum+=func1(i)
                else:
                        sum+=i
        return sum

永恒的蓝色梦想

高义校长 发表于 2020-4-2 16:31
不好意思，发太快，就是想问如何实现。

问题2

from itertools import zip_longest

def func2(*arrays):
        return [sum(i) for i in zip_longest(*arrays,fillvalue=0)]

高义校长

永恒的蓝色梦想 发表于 2020-4-2 16:39
问题2

非常感谢您。我是一个新手，以前会一点简单的VBA。
您这个是引入新的模块。
请教一下，python是否有数组下标类似的概念：
如：cr(i)=ar(i)+br(i)。

永恒的蓝色梦想

高义校长 发表于 2020-4-2 16:46
非常感谢您。我是一个新手，以前会一点简单的VBA。
您这个是引入新的模块。
请教一下，python是否有数 ...

有，比如有这样一个列表

a=[1,2,3,4,5]

，

a[4]

就可以访问到第5个值，也就是

5

高义校长

永恒的蓝色梦想 发表于 2020-4-2 16:47
有，比如有这样一个列表，就可以访问到第5个值，也就是

那第二个问题是不是可以直接相加。但我这边就提示索引越界了。

永恒的蓝色梦想

高义校长 发表于 2020-4-2 16:50
那第二个问题是不是可以直接相加。但我这边就提示索引越界了。

不能，两个数组不一样长，直接加肯定越界

给你一个不导入模块的版本

def func2(*its):
    its=map(iter,its)
    temp=[]

    while True:
        sum=0
        flag=True

        for i in its:
            try:
                sum+=next(i)
                flag=False

            except StopIteration:
                pass

        if flag:
            return temp
        
        temp.append(sum)

高义校长

永恒的蓝色梦想 发表于 2020-4-2 16:53
不能，两个数组不一样长，直接加肯定越界

给你一个不导入模块的版本

感谢感谢，我还得消化一下。如果出现了3维或以上，是不是比较麻烦。

永恒的蓝色梦想

高义校长 发表于 2020-4-2 16:55
感谢感谢，我还得消化一下。如果出现了3维或以上，是不是比较麻烦。

递归 就是应对这种问题的，感兴趣可以搜索一下

高义校长

永恒的蓝色梦想 发表于 2020-4-2 16:59
递归 就是应对这种问题的，感兴趣可以搜索一下

前面那个单纯的一维列表我研究出来了，但二级及以上我就研究不清楚了，求助。
第一个问题：
数组所有元素求和：
a=[1, 2, 3, 4, 5, 1, 2, 3, 1, 2, 3, [1, 2, [1,2,3,5]]]
def sm(x):
    return sum(i in x if isinstance(i,int) else sm(i) for i in x)
print(sm(a))
第二个问题就不行了，为什么我第二个问题会报错呢。
a=[[1,2,3,4],[1,2],[1,2,3],5]
b=[[1,2,3],[1,2,3],[1,2],5,6]
from itertools import zip_longest
class Arsum:
    def __init__(self,m,n):
        self.m=m
        self.n=n
    def getlst(self):
        return [sum(x) for x in zip_longest(self.m,self.n,fillvalue=0)]
c=Arsum(a,b)
c.getlst()

高义校长

有没有大佬再帮忙看一下，怎么计算。

永恒的蓝色梦想

高义校长 发表于 2020-4-4 20:35
前面那个单纯的一维列表我研究出来了，但二级及以上我就研究不清楚了，求助。
第一个问题：
数组所有元 ...

不报错就奇怪了，sum的列表里既有list还有int
还有，没事用什么类

from itertools import zip_longest

def flatten(iterable,/):
    for element in iterable:
        if hasattr(element,'__iter__'):
            yield from flatten(element)
        
        else:
            yield element

def Arsum(*v):
    return [sum(i) for i in zip_longest(*map(flatten,v),fillvalue=0)]

a=[[1,2,3,4],[1,2],[1,2,3],5]
b=[[1,2,3],[1,2,3],[1,2],5,6]

print(Arsum(a,b))

PS:提问新内容记得开新帖，

高义校长

永恒的蓝色梦想 发表于 2020-4-4 22:51
不报错就奇怪了，sum的列表里既有list还有int
还有，没事用什么类PS:提问新内容记得开新帖，

非常感谢。