如何在python中转置3D列表?

时间:2016-02-07 23:56:13

标签: python numpy matrix

假设我有尺寸为9 x 9 x 26的矩阵m

[[['a00', ... 'z00'], ['a01', ... 'z01'], ... ['a09', ... 'z09']],
[['a10', ... 'z10'], ['a11', ... 'z11'], ... ['a19', ... 'z19']],
...
[['a90', ... 'z90'], ['a91', ... 'z91'], ... ['a99', ... 'z99']]]

我想将A转换为以下尺寸为26 x 81的矩阵:

 [['a00', 'a01', ... 'a09', 'a10', ... 'a19', ... 'a99'],
  ['z00', 'z01', ... 'z09', 'z10', ... 'z19', ... 'z99']]

在python中执行此操作的最佳方法是什么?

2 个答案:

答案 0 :(得分:3)

这是你想要的那种重新排序吗?

In [128]: a=[[[11,12,13],[14,15,16]],[[21,22,23],[24,25,26]]]

In [129]: A=np.array(a)

In [130]: A
Out[130]: 
array([[[11, 12, 13],
        [14, 15, 16]],

       [[21, 22, 23],
        [24, 25, 26]]])

In [131]: A.reshape(-1,3).T
Out[131]: 
array([[11, 14, 21, 24],
       [12, 15, 22, 25],
       [13, 16, 23, 26]])

A(2,2,3),结果为(3,4)

我将其重新塑造为(4,3),然后转换了2个轴。

有时像这样的问题会让人感到困惑,特别是如果他们没有具体的例子来测试。如果我没有做到这一点,我需要首先应用转置,并提供完整的规范。

e.g。

In [136]: A.transpose([2,0,1]).reshape(3,-1)
Out[136]: 
array([[11, 14, 21, 24],
       [12, 15, 22, 25],
       [13, 16, 23, 26]])

如果您有列表,而不是数组,则任务将更加困难。有一个简单的习惯用于转置' a' 2d'嵌套列表,但它可能不适用于' 3d'。

In [137]: zip(*a)
Out[137]: [([11, 12, 13], [21, 22, 23]), ([14, 15, 16], [24, 25, 26])]

(好的,另一个答案是非阵列解决方案)。

答案 1 :(得分:2)

如果你有一个列表而不是一个numpy数组:

m = [[['a00',  'z00'], ['a01',  'z01'],  ['a09',  'z09']],
[['a10',  'z10'], ['a11',  'z11'],  ['a19',  'z19']],   
[['a90',  'z90'], ['a91',  'z91'],  ['a99',  'z99']]]

from itertools import chain
print zip(*(chain(*m))

[('a00', 'a01', 'a09', 'a10', 'a11', 'a19', 'a90', 'a91', 'a99'), 
('z00', 'z01', 'z09', 'z10', 'z11', 'z19', 'z90', 'z91', 'z99')]