[Python]リスト(List)の形状を取得するには？

どうも、ちょげ(@chogetarou)です。

リストの形状を取得する方法を紹介します。

方法

リストの形状を取得する方法は、２つあります。

len()

１つは、len()を使う方法です。

まず、len()を呼び出し、引数にリストを指定します。

そして、もう1度len()を呼び出し、引数にリストのインデックス「０」の要素を指定します。

rows = len(list) #行の数
columns = len(list[0]) #列の数

引数にリストを指定したlen()で行数、引数にリストのインデックス「０」を指定したlen()で列の数を取得できます。

使用例

numbers = [[1, 2], [3, 4], [5,6]]

rows = len(numbers)
columns = len(numbers[0])

print('行：{},列：{}'.format(rows, columns)) #行：3,列：2

numpy.shape()

もう１つは、numpyのshape()を使う方法です。

まず、numpyをインポートします。

import numpy

そして、numpyからshape()を呼び出します。

shape()の引数にリストを指定します。

numpy.shape(list)

numpyのshape()は、引数に指定したリストの形状を返します。

使用例

import numpy

numbers = [[1, 2], [3, 4], [5,6]]

shape = numpy.shape(numbers)

print(shape) #(3, 2)

まとめ

リスト(List)の形状を取得する方法は、次の２つです。

• len()を使う方法
• numpy.shape()を使う方法