为什么笛卡尔积会产生“ TypeError:在0维数组上的迭代”?

时间:2019-05-31 00:13:56

标签: python itertools cartesian-product

我正在尝试从数组列表中制作笛卡尔积,但是它一直给我一个TypeError: iteration over a 0-d array

我有一个看起来像这样的列表

print(a)
>>>[array([1., 2.]), array([3., 4.]), array(1400.)]

现在,当我尝试这样做时:

b=list(itertools.product(*a))
>>>TypeError: iteration over a 0-d array

我想念什么?

1 个答案:

答案 0 :(得分:2)

正如@ user2357112在评论中所解释的,您当前拥有的最后一个元素为0维数组。如果检查它的长度,将得到TypeError: len() of unsized object。为了使您的解决方案有效,您需要使用[]将元素封装在最后一个数组中,以便能够使用product

import itertools

a = [np.array([1., 2.]), np.array([3., 4.]), np.array([1400.])]

b = list(itertools.product(*a))

#[(1.0, 3.0, 1400.0),
# (1.0, 4.0, 1400.0),
# (2.0, 3.0, 1400.0),
# (2.0, 4.0, 1400.0)]

编辑,根据要求回答第二个问题:

import itertools

dict1 = {'wdth_i': ['1', '2'], 'wdth_p': ['3', '4'], 'mu': '1400'}

a = [] 
for i in dict1.values():
    if isinstance(i, list):
        a.append(i)  
    else:
        a.append([i])

f = list(itertools.product(*a))
# [('1', '3', '1400'),
#  ('1', '4', '1400'),
#  ('2', '3', '1400'),
#  ('2', '4', '1400')]