web-dev-qa-db-ja.com

Pythonで行列を転置する

Pythonで行列転置関数を作成しようとしています。マトリックスは、整数のリストのリストとして表される2次元配列です。たとえば、次は2X3マトリックスです（マトリックスの高さが2で幅が3であることを意味します）：

A=[[1, 2, 3],
   [4, 5, 6]]

転置するには、i番目のインデックスのj番目のアイテムがj番目のインデックスのi番目のアイテムになります。上記のサンプルがどのように転置されるかを次に示します。

>>> transpose([[1, 2, 3],
               [4, 5, 6]])
[[1, 4],
[2, 5],
[3, 6]]
>>> transpose([[1, 2],
               [3, 4]])
[[1, 3],
[2, 4]]

これどうやってするの？

pythonlistlist-comprehension
27
Asher Garland

Zip を*とともに使用して、行列の転置を取得できます。

>>> A = [[ 1, 2, 3],[ 4, 5, 6]]
>>> Zip(*A)
[(1, 4), (2, 5), (3, 6)]
>>> lis  = [[1,2,3], 
... [4,5,6],
... [7,8,9]]
>>> Zip(*lis)
[(1, 4, 7), (2, 5, 8), (3, 6, 9)]

返されるリストをリストのリストにする場合：

>>> [list(x) for x in Zip(*lis)]
[[1, 4, 7], [2, 5, 8], [3, 6, 9]]
#or
>>> map(list, Zip(*lis))
[[1, 4, 7], [2, 5, 8], [3, 6, 9]]
100
Ashwini Chaudhary

怠け者であり、NumPy配列の転置機能を使用することに対する賞はありますか？ ;）

import numpy as np

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

b = a.transpose()
23
Klaus-Dieter Warzecha

同じ行列を返したい場合は、次のように記述します。

return [[ m[row][col] for col in range(0,width) ] for row in range(0,height) ]

これは、各行を調べて各列の各要素を返すことにより、行列mを反復処理します。したがって、順序は次のようになります。

[[1,2,3],
[4,5,6],
[7,8,9]]

質問3では、代わりに列ごとに行き、各行の各要素を返します。したがって、順序は次のようになります。

[[1,4,7],
[2,5,8],
[3,6,9]]

したがって、反復する順序を変更するだけです。

return [[ m[row][col] for row in range(0,height) ] for col in range(0,width) ]
11
Asher Garland