2次元配列リストにおける[a, b]→[b, a]変換について

Question

a
array([[278, 668],
       [278, 669],
       [278, 670],
       ...,
       [748, 686],
       [748, 687],
       [748, 688]])

b
array([[668, 278],
       [669, 278],
       [670, 278],
       ...,
       [686, 748],
       [687, 748],
       [688, 748]])

上記のaの配列を読み込んだ時にbを出力するプログラムを作成したいです。
ここでa[0], a[1] = a[1], a[0]のようなプログラムを試しましたが、うまくいきませんでした。
宜しくお願いします。

Answer 1

これは縦方向のスライスなのでa[:,0]等と記述します。

https://docs.scipy.org/doc/numpy/user/quickstart.html#indexing-slicing-and-iterating

問題はそのスライスで同様に書いてもうまくいかないこと。

>>> import numpy as np
>>> a = np.arange(8).reshape(4, 2)  # データ生成
>>> a  # 確認
array([[0, 1],
       [2, 3],
       [4, 5],
       [6, 7]])
>>> a[:,0], a[:,1] = a[:,1], a[:,0]
>>> a
array([[1, 1],
       [3, 3],
       [5, 5],
       [7, 7]])

右辺をコピーするとできます。

>>> a = np.arange(8).reshape(4, 2)  # データ生成
>>> a[:,0], a[:,1] = a[:,1].copy(), a[:,0].copy()
>>> a
array([[1, 0],
       [3, 2],
       [5, 4],
       [7, 6]])

しかし、fancy indexを使ってこう書いた方が良いかもしれません。

>>> a = np.arange(8).reshape(4, 2)  # データ生成
>>> a[:,[1,0]]
array([[1, 0],
       [3, 2],
       [5, 4],
       [7, 6]])

当然再代入が必要になります。

Comments

[Kohei_KESE]

回答ありがとうございます。
無事問題を解決することができましたし、referenceで勉強にもなりました。

Answer 2

参考情報

• NumPy配列ndarrayの行・列を任意の順番に並べ替え、選択（抽出）

https://note.nkmk.me/python-numpy-swap-select/

列の順番を入れ替える方法と、 行を逆順にする方法で試しみました。

実行例

Answer 3

array = [[1,5],[2,3],[4,8],[7,4],[1,4]]

for i in range(len(array)):
    tmp = array[i][0]
    array[i][0] = array[i][1]
    array[i][1] = tmp
    
for number in array:
    print(number)

Answer 4

こういうことでしょうか？

a = [[i * 2, i * 2 + 1]for i in range(4)]
print(a)

for i in a:
    i[0],i[1] = i[1],i[0]
print(a)

出力結果

3.6.6 |Anaconda, Inc.| (default, Jun 28 2018, 11:27:44) [MSC v.1900 64 bit (AMD64)]
[[0, 1], [2, 3], [4, 5], [6, 7]]
[[1, 0], [3, 2], [5, 4], [7, 6]]