[已解决]关于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

1. def func1(array):
   
2.         sum=0
   
3.         for i in array:
   
4.                 if type(i) is list:
   
5.                         sum+=func1(i)
   
6.                 else:
   
7.                         sum+=i
   
8.         return sum

复制代码

跳转到最佳答案楼层

永恒的蓝色梦想

问题呢？？？

高义校长

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

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

永恒的蓝色梦想

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

问题1

1. def func1(array):
   
2.         sum=0
   
3.         for i in array:
   
4.                 if type(i) is list:
   
5.                         sum+=func1(i)
   
6.                 else:
   
7.                         sum+=i
   
8.         return sum

复制代码

永恒的蓝色梦想

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

问题2

1. from itertools import zip_longest
   
2. 
   
3. def func2(*arrays):
   
4.         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是否有数 ...

有，比如有这样一个列表

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

复制代码

，

1. a[4]

复制代码

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

1. 5

复制代码

高义校长

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

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

永恒的蓝色梦想

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

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

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

1. def func2(*its):
   
2.     its=map(iter,its)
   
3.     temp=[]
   
4. 
   
5.     while True:
   
6.         sum=0
   
7.         flag=True
   
8. 
   
9.         for i in its:
   
10.             try:
    
11.                 sum+=next(i)
    
12.                 flag=False
    
13. 
    
14.             except StopIteration:
    
15.                 pass
    
16. 
    
17.         if flag:
    
18.             return temp
    
19.         
    
20.         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
还有，没事用什么类

1. from itertools import zip_longest
   
2. 
   
3. def flatten(iterable,/):
   
4.     for element in iterable:
   
5.         if hasattr(element,'__iter__'):
   
6.             yield from flatten(element)
   
7.         
   
8.         else:
   
9.             yield element
   
10. 
    
11. def Arsum(*v):
    
12.     return [sum(i) for i in zip_longest(*map(flatten,v),fillvalue=0)]
    
13. 
    
14. a=[[1,2,3,4],[1,2],[1,2,3],5]
    
15. b=[[1,2,3],[1,2,3],[1,2],5,6]
    
16. 
    
17. print(Arsum(a,b))

复制代码

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

高义校长

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

非常感谢。