【NumPy】配列の大きさ（行数・列数）を取得

この記事では、Python言語とNumPyを用いて、配列の大きさ（行数・列数）を取得する方法をソースコード付きで解説します。

配列の大きさ（行数・列数）

Pythonの数値計算モジュールNumPyでは、numpy.shapeメソッドで配列の大きさ（行数・列数）を取得できます。

ソースコード

サンプルプログラムのソースコードです。

# -*- coding: utf-8 -*-
import numpy as np

# 2次元配列の宣言・初期化
A = np.array([[1, 2],
              [3, 4],
              [5, 6]])

# 行列の大きさ
print("行列Aの大きさ:", A.shape)
print("行列Aの行数:", A.shape[0])
print("行列Aの列数:", A.shape[1])

"""
行列Aの大きさ:(3, 2)
行列Aの行数:3
行列Aの列数:2
"""

おすすめ記事

Python入門 サンプル集
NumPy入門 サンプル集