[已解决]数组计算问题（新）

高义校长

请问各位大佬：
有2个数组：
a=[[1,2,3,4],[1,2],[1,2,3],5]
b=[[1,2,3],[1,2,3],[1,2],5,6]

怎么能得到数组c:
[[2,4,6,4],[2,4,3],[2,4,3],10,6]

数组c也可以变成
[[2,4,6,4],[2,4,3],[2,4,3],[10],[6]]
能否两种都给写一代码，非常感谢。

最佳答案

月排行榜 / 总排行榜

sunrise085

2020-4-4 22:26:32

1. def fun(a,b):
   
2.         c=[]
   
3.         if len(a)<=len(b):
   
4.                 a,b=b,a
   
5.         for i in range(len(a)):
   
6.                 if i<len(b):
   
7.                         if type(a[i])==list:
   
8.                                 c.append(fun(a[i],b[i]))
   
9.                         else:
   
10.                                 c.append(a[i]+b[i])
    
11.                 else:
    
12.                         c.append(a[i])
    
13.         return c
    
14. 
    
15. a=[[1,2,3,4],[1,2],[1,2,3],5]
    
16. b=[[1,2,3],[1,2,3],[1,2],5,6]
    
17. print(fun(a,b))

复制代码

跳转到最佳答案楼层

倒戈卸甲

感觉步骤有点多啊，不想动手=_=

高义校长

倒戈卸甲 发表于 2020-4-4 22:01
感觉步骤有点多啊，不想动手=_=

辛苦一下啊，哈哈。我看他们用帝归不是2行就搞定了啊。我不会，大佬费心。

高义校长

倒戈卸甲 发表于 2020-4-4 22:01
感觉步骤有点多啊，不想动手=_=

其实前面我问过类似的问题，大佬给出了答案，但是我感觉我悟性不够啊。
https://fishc.com.cn/thread-163560-1-1.html

sunrise085

1. def fun(a,b):
   
2.         c=[]
   
3.         if len(a)<=len(b):
   
4.                 a,b=b,a
   
5.         for i in range(len(a)):
   
6.                 if i<len(b):
   
7.                         if type(a[i])==list:
   
8.                                 c.append(fun(a[i],b[i]))
   
9.                         else:
   
10.                                 c.append(a[i]+b[i])
    
11.                 else:
    
12.                         c.append(a[i])
    
13.         return c
    
14. 
    
15. a=[[1,2,3,4],[1,2],[1,2,3],5]
    
16. b=[[1,2,3],[1,2,3],[1,2],5,6]
    
17. print(fun(a,b))

复制代码

高义校长

sunrise085 发表于 2020-4-4 22:26

非常感谢您。
能否请教一下，您这个能扩展吗？
如果有a,b,c,d或者N多列表呢

sunrise085

高义校长 发表于 2020-4-4 22:33
非常感谢您。
能否请教一下，您这个能扩展吗？
如果有a,b,c,d或者N多列表呢

没太明白，N多列表是啥意思
意思是不一定只有a b,也可能是a b c d……合成m
是这个意思吗？
那样的话，这个函数肯定不行，这个函数这不是只有两个参数么。
需要在此基础之上进行修改完善。
我是在手机上码字的 ，很不方便码代码，你自己按照这个基础版本需要一下吧。或者等其他人回答吧。

高义校长

sunrise085 发表于 2020-4-4 22:42
没太明白，N多列表是啥意思
意思是不一定只有a b,也可能是a b c d……合成m
是这个意思吗？

非常感谢您。
https://fishc.com.cn/thread-163560-1-1.html
您这个也能解决我的问题的，就是改起来比较麻烦。
其实我是想用这个写个递归，然后就可以扩展了，但是就是始终写不对。
[x for x in zip_longest(self.m,self.n,fillvalue=0)]

sunrise085

高义校长 发表于 2020-4-4 22:52
非常感谢您。
https://fishc.com.cn/thread-163560-1-1.html
您这个也能解决我的问题的，就是改起来比 ...

这里也用到递归了啊，第八行，若该元素是list则递归