numpy.random.permutation – 配列の要素をランダムに並べ替えた新しい配列を生成

numpy.random.permutationは、渡した配列の要素をランダムに並べ替える関数です。多次元配列を渡した場合は最初の軸だけを並べ替えます。

なお、同じような関数にnumpy.random.shuffleがあります。両者の違いは、permutationは渡した配列の要素を並べ替えた新しい配列を生成するのに対して、shuffleは引数に渡した配列の要素を並べ替える点にあります。

このページでは、random.permutation関数について解説していきます。

1. 書式

np.random.permutation(x)

2. サンプルコード

引数に整数を与えた場合は、np.arange(x)の配列の要素をランダムに並べ替えます。np.arangeは『numpy.arange – 色々な連番の配列を生成する』で解説しています。

import numpy as np
np.random.permutation(10)
# >>> np.arange(x)を渡すのと同じ

array([2, 4, 0, 1, 7, 9, 3, 5, 8, 6])

リストやタプルなどの配列オブジェクト(array_like)を渡すこともできます。

np.random.permutation([1, 3, 5, 7, 11, 13])

array([ 7,  3, 11, 13,  5,  1])

多次元配列を渡した場合は、最初の軸に沿って並べ替えます。たとえば2次元配列を渡すと、行をランダムに並べ替えます。

# 2次元配列を渡すと行を並べ替える。
arr = np.arange(9).reshape((3, 3))
np.random.permutation(arr)

array([[6, 7, 8],
       [0, 1, 2],
       [3, 4, 5]])

コードは書きませんが、3次元配列の場合は奥行きのみをランダムに並べ替えます。

なお、Generator.permutationではオプション引数 axis が追加され、多次元配列を渡した場合に、並べ替える軸を指定することが可能になっています。

3. まとめ

以上、numpy.random.permutationは、配列の要素をランダムに並べ替える関数です。現在は、この関数は使わず、Generator.permutationというジェネレータメソッドを使います。

このような変更が行われた最大の理由は、処理の高速化にあります。科学技術計算では、大量のデータを扱うため高速性が必須なのです。

今後はGenerator.permutationで一本化するようにしましょう。