[Python] NumPy の使い方

PythonのNumPyは、数値計算を行うための高速で効率的なライブラリです。NumPyは、多次元配列をサポートし、配列操作や数学関数、線形代数の計算などの多くの機能を提供します。NumPyの使い方について、以下に詳しく説明していきます。

目次

NumPyのインストール

まず、NumPyを使用するには、まずNumPyをインストールする必要があります。NumPyをインストールするには、ターミナルで以下のコマンドを実行します。

pip install numpy

NumPyの基本的なデータ構造

NumPyでは、多次元配列(ndarray)を使用します。これは、リストやタプルと似たようなデータ構造で、要素の型が同じであることが必要です。NumPy配列を作成するには、np.array()関数を使用します。たとえば、以下のようにしてNumPy配列を作成することができます。

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr)

この場合、出力結果は以下のようになります。

[1 2 3 4 5]

また、2次元配列を作成するには、以下のようにします。

import numpy as np

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

この場合、出力結果は以下のようになります。

[[1 2 3]
 [4 5 6]]

NumPy配列の属性

NumPy配列には、多くの属性があります。以下に、NumPy配列の主要な属性をいくつか紹介します。

  • ndim: 配列の次元数を返します。
  • shape: 配列の各次元の大きさをタプルで返します。
  • size: 配列の要素数を返します。
  • dtype: 配列の要素のデータ型を返します。

たとえば、以下のコードを実行すると、NumPy配列の属性を確認することができます。

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print("配列の次元数:", arr.ndim)
print("配列の大きさ:", arr.shape)
print("配列の要素数:", arr.size)
print("配列のデータ型:", arr.dtype)

この場合、出力結果は以下のようになります。

配列の次元数: 2
配列の大きさ: (2, 3)
配列の要素数: 6
配列のデータ型: int64

NumPy配列の操作

NumPy配列は、Pythonの標準リストと似たような方法で操作することができます。以下に、NumPy配列の主要な操作をいくつか紹介します。

インデックスとスライス

NumPy配列は、リストと同じようにインデックスやスライスで要素にアクセスすることができます。たとえば、以下のようにしてNumPy配列の一部分を取得することができます。

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr[2])       # 3
print(arr[1:4])     # [2 3 4]

形状の変更

NumPy配列の形状を変更するには、reshape()関数を使用します。たとえば、以下のようにして、1次元配列を2次元配列に変換することができます。

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])
new_arr = arr.reshape(2, 3)
print(new_arr)

この場合、出力結果は以下のようになります。

[[1 2 3]
 [4 5 6]]

転置

NumPy配列を転置するには、T属性を使用します。たとえば、以下のようにして、2次元配列を転置することができます。

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
new_arr = arr.T
print(new_arr)

この場合、出力結果は以下のようになります。

[[1 4]
 [2 5]
 [3 6]]

配列の結合

NumPy配列を結合するには、concatenate()関数を使用します。たとえば、以下のようにして、2つの配列を横方向に結合することができます。

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6]])
new_arr = np.concatenate((arr1, arr2), axis=0)
print(new_arr)

この場合、出力結果は以下のようになります。

[[1 2]
 [3 4]
 [5 6]]

NumPyの数学関数

NumPyは、多くの数学関数を提供しています。以下に、NumPyの主要な数学関数をいくつか紹介します。

  • sqrt(): 平方根を計算します。
  • exp(): 指数関数を計算します。
  • sin(), cos(), tan(): 三角関数を計算します。
  • log(), log10(): 対数関数を計算します。
  • sum(): 配列の合計を計算します。
  • mean(): 配列の平均を計算します。
  • std(): 配列の標準偏差を計算します。
  • min(), max(): 配列の最小値、最大値を計算します。

以下に、これらの関数を使用した例をいくつか紹介します。

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(np.sqrt(arr))             # [1.         1.41421356 1.73205081 2.         2.23606798]
print(np.exp(arr))              # [  2.71828183   7.3890561   20.08553692  54.59815003 148.4131591 ]
print(np.sin(arr))              # [ 0.84147098  0.90929743  0.14112001 -0.7568025  -0.95892427]
print(np.sum(arr))              # 15
print(np.mean(arr))             # 3.0
print(np.std(arr))              # 1.4142135623730951
print(np.min(arr), np.max(arr)) # 1 5

NumPyのブロードキャスト

NumPyのブロードキャストとは、2つ以上の配列に対して、要素ごとの演算を自動的に適用する機能です。以下に、ブロードキャストの例をいくつか紹介します。

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
print(arr1 + arr2)          # [5 7 9]

arr3 = np.array([[1, 2, 3], [4, 5, 6]])
arr4 = np.array([1, 2, 3])
print(arr3 + arr4)          # [[2 4 6]
                            #  [5 7 9]]

arr5 = np.array([[1], [2]])
arr6 = np.array([1, 2, 3])
print(arr5 + arr6)          # [[2 3 4]
                            #  [3 4 5]]

NumPyのデータの保存と読み込み

NumPyは、配列データをバイナリ形式で保存する機能を提供しています。以下に、NumPy配列をファイルに保存する方法と、保存されたファイルからNumPy配列を読み込む方法を紹介します。

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
np.save('my_array.npy', arr)     # 配列をファイルに保存する

new_arr = np.load('my_array.npy') # ファイルから配列を読み込む
print(new_arr)                   # [1 2 3 4 5]

まとめ

以上が、NumPyの主な機能や使用方法についての説明です。NumPyは、高速で効率的な数値演算ができるように設計されたライブラリで、多次元配列を扱うことができます。NumPyの配列は、Pythonのリストよりも効率的で、多くの便利な関数やメソッドを提供しています。

NumPyの配列を作成する方法には、以下のようなものがあります。

  • numpy.array(): Pythonのリストから配列を作成します。
  • numpy.zeros(): 全ての要素が0の配列を作成します。
  • numpy.ones(): 全ての要素が1の配列を作成します。
  • numpy.arange(): 指定された範囲の整数の配列を作成します。
  • numpy.random.rand(): 0から1までの一様分布からランダムな数値を生成して配列を作成します。

NumPyの配列を操作するためのいくつかの関数とメソッドを紹介しました。これらの関数とメソッドは、NumPyの配列に対して、様々な数値演算を行うためのものです。

NumPyのブロードキャスト機能を使うことで、複数の配列に対して要素ごとの演算を自動的に適用することができます。

最後に、NumPyの配列をファイルに保存する方法と、保存されたファイルから配列を読み込む方法を紹介しました。

NumPyは、科学技術計算やデータ分析などの分野で広く使われているライブラリであり、Pythonで数値計算を行う際には欠かせないものです。

よかったらシェアしてね!
  • URLをコピーしました!
  • URLをコピーしました!
目次