为什么笛卡尔积会产生“TypeError：0-d 数组上的迭代”？

Question

我正在尝试从数组列表中制作笛卡尔积，但它一直给我一个 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

我错过了什么？

Answer 1

正如 @user2357112 在 cmets 中所解释的，您当前将最后一个元素作为 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')]