假设我有一个NumPy数组,如下所示:
0 1 (2,3)
(4,5) 6 7
9 (10,) 11
我如何将其拆分为以下数组的列表?条件:最终数组中没有元组,并且每个数组都包含元组元素的唯一组合。也就是说,原始数组中有两个元组,因此这些元素有4种不同的组合。下面的第一个数组具有原始数组的组合(2)和(4)。
0 1 2
4 6 7
9 10 11
0 1 2
5 6 7
9 10 11
0 1 3
4 6 7
9 10 11
0 1 3
5 6 7
9 10 11
如@unutbu所建议的,这样的数据结构很尴尬。但是,如果您想这样做,那是可行的:
In [52]:
A
Out[52]:
array([[0, 1, (2, 3)],
[(4, 5), 6, 7],
[9, (10,), 11]], dtype=object)
In [53]:
import itertools
AR=A.ravel()
AT=[i for i, val in enumerate(AAR) if isinstance(val, tuple)]
for item in itertools.product(*AR[AT]):
B=AR.copy()
B[AT]=item
print B.reshape(A.shape)
[[0 1 2]
[4 6 7]
[9 10 11]]
[[0 1 2]
[5 6 7]
[9 10 11]]
[[0 1 3]
[4 6 7]
[9 10 11]]
[[0 1 3]
[5 6 7]
[9 10 11]]
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句