【NumPy】配列の大きさ(行数・列数)を取得

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

配列の大きさ(行数・列数)

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

ソースコード

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

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

def main():

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

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


if __name__ == '__main__':
    main()

実行結果

サンプルプログラムの実行結果です。

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

おすすめ記事

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

シェア&フォローお願いします!