我在列表中有数组,我想从两个不同的列表中计算数组中元素的数量,而不是计算列表项的数量。你知道吗
代码
import numpy as np
def count_total(a,b):
#count the total number of element for two arrays in different list
x,y=len(a),len(b)
result=[]
for a1 in a:
for b2 in b:
result.append(x+y)
return result
a=[np.array([2,2,1,2]),np.array([1,3])]
b=[np.array([4,2,1])]
c=[np.array([1,2]),np.array([4,3])]
print(count_total(a,b))
print(count_total(a,c))
print(count_total(b,c))
实际产量
[3, 3]
[4, 4, 4, 4]
[3, 3]
所需输出
[7,5]
[6,6,4,4]
[5,5]
有人能帮忙吗?你知道吗
在我看来,从你们的例子来看,你们需要所有可能的方法来求数组的长度之和。这可以通过
itertools.product
实现。这是我的密码:产品返回a和b中每个元素的所有可能排列。然后对于每个排列,我们从每个列表中获取排列中的部分的长度,然后将它们与
sum
相加。你知道吗错误在第4行,x和y被分配列表长度而不是数组长度。你知道吗
更换管路4-8
与
相关问题 更多 >
编程相关推荐