NEW ARTICLE
PyCharm 最近の変更を一覧表示するショートカットキー
  1. HOME
  2. python
  3. python numpyの多次元配列の次元数を取得する

python numpyの多次元配列の次元数を取得する

python numpyの多次元配列の次元数を取得する

pythonで、ライブラリnumpyのndimを使用して、多次元配列の次元数を取得するサンプルコードを記述してます。pythonのバージョンは3.8.5を使用してます。

目次

環境

  • OS windows10 pro 64bit
  • python 3.8.5

numpyインストール

numpyをインストールされていない方は、pipでインストールしておきます。

pip install numpy

# Successfully installed numpy-1.19.4

ndim使い方

ndimを使用すると、多次元配列の次元数を取得することが可能です。

import numpy as np

arr = np.array([["a", "b", "c"], ["d", "e", "f"]])

print(arr.ndim)
# 2

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

print(arr.ndim)
# 3

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

print(arr.ndim)
# 4

# 1次元の場合は1となります
arr = np.array([1, 2, 3])

print(arr.ndim)
# 1