初心者のためのDjango入門

■第15話：NumPyの便利な機能

（最終更新日：2023.06.11）

「NumPy配列を自由自在に操ろう！」

これまで、行列計算から統計計算、DeepLearningまで、幅広い領域でNumPyが便利に使えることを説明してきた。 今回は、今まで紹介してこなかったNumPyの便利機能について紹介したいと思う。

配列の結合と分離、条件付き演算というフィルタリング的な機能まで、NumPyは様々な操作に対応する。 ぜひ、今からでも使えるようになろう。

１.配列の結合と分割

NumPyには、配列を結合する関数と分割する関数が容易されている。 np.concatenateは結合する関数で、使い方は以下の通りになる。 2つの配列が結合されて、[1, 2, 3, 4, 5, 6]になる。

また、分割する関数はnp.splitで、使い方は以下の通りになる。 元の配列が3分割されて[1,2],[3,4],[5,6]になる。

２.条件付き演算

NumPyには条件付き演算を行う関数が用意されている。 それがnp.whereで、NumPy配列内の要素を選択するための便利な関数である。 下記では、4を超える値を選択しており、フィルタされて配列[5, 6]になっている。

以上のように、NumPyはPythonの科学計算とデータ分析のエコシステムにおいて、中心的な役割を果たしている。 これらのライブラリは互いに深く関連しており、一緒に使用されることで、強力なデータ分析ツールセットを形成している。 同様の操作が第13話で紹介したmaskingでも可能である。 実際は中の挙動が異なるのだが、どちらもある条件を満たす要素を抽出すると考えて良いだろう。

３.ファンシーインデックス

ファンシーインデックス（または整数配列インデックス）は、NumPy配列に対して特定の順序でインデックスを指定するための機能である。 これにより、複雑な配列構造を生成したり、配列の部分集合を特定の順序で抽出したりすることができる。 以下にその使用例を示す。NumPy配列とインデックス配列をそれぞれ作成し、ファンシーインデックスという手法で、 インデックス配列で指定した要素のみを取り出している。下記では1,3,4番目の要素を抽出し、[20, 40, 50]が取得された。

この操作が凄いのは、多次元配列にも適用できるところである。 取得したい行要素と列要素をそれぞれNumPy配列で指定することで、多次元配列から選択的に要素を抽出できる。 下記では、1行2列目の要素と2行2列目の要素を抽出（0行、0列から数えるものとする）している。結果、[5, 9]が抽出された。

４.ユニバーサル関数

ユニバーサル関数はNumPyが提供する機能の一つで、配列の要素ごとに操作を行う関数のことを指す。 これにより、ループを書くことなく、配列の全要素に対する操作を効率的かつ簡潔に書くことができる。 これは、Pythonの組込関数（例えば、add、subtract、multiplyなど）が一度に一つの要素しか処理できないのと対照的である。

NumPyのユニバーサル関数は、C言語レベルで実装されているため、 Pythonの組み込み関数よりも大幅に高速だ。 これは「ベクトル化」と呼ばれる手法の一種で、コードの可読性とパフォーマンスの両方を向上させることができる。 以下に、ユニバーサル関数の基本的な使い方を示す。 各要素を2乗するsquare、各要素に値を加えるadd、各要素を逆数にするreciprocal等がある。 reciprocalは値が小数になるので、型をfloatにしている。一気に計算できているのが分かるだろう。

５.まとめ

今回、NumPyで今まで説明してこなかった便利な関数についてまとめてみた。 NumPyは値がベクトル化されていて、一気に高速に計算処理できるというのが、圧倒的な強みである。 しかも、コードの可読性に優れ、比較的簡単なコードで計算できるのも大きい。 少なくとも、今回説明したユニバーサル関数と13話で説明したブロードキャスティングだけは使えるようになろう。

全5回に続いたNumPyシリーズも今回で終了だが、勉強だけにとどまらずぜひ、実務でも活用してみよう。 きっと、様々な場面で役に立つはずだ。

▼参考図書、サイト

  NumPyで条件に応じた処理を行うnp.whereの使い方 　note.nkmk.me
  NumPyのファンシーインデックス（リストによる選択と代入） 　note.nkmk.me
  NumPy入門 代表的なユニバーサル関数 　PythonエンジニアによるPython3学習サイト