旅行好きなソフトエンジニアの備忘録

プログラミングや技術関連のメモを始めました

【Python】How to swap two columns in a 2d numpy array? - 101 Numpy Exercises

Q:

変数arrのカラム0とカラム1を入れ替えなさい

arr = np.arange(9).reshape(3,3)
#> array([[0, 1, 2],
#>        [3, 4, 5],
#>        [6, 7, 8]])

A:

arr = arr[:, [1, 0, 2]]
#> array([[1, 0, 2],
#>        [4, 3, 5],
#>        [7, 6, 8]])


www.machinelearningplus.com