我正在尝试从数组列表中制作笛卡尔积,但是它一直给我一个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
我想念什么?
答案 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')]