【毎日Python】Pythonで全ての要素が0や1の配列を作成する方法|numpy.zeros/ones

zeros
毎日Python/こつこつPython

Pythonで全ての要素が0または1の配列を作成する方法です。
使用するのは、Pythonのnumpyライブラリの、zeros関数、ones関数です。

まず、全ての値が0の配列を作成してみましょう。
zeros関数の引数に、要素数を指定します。
実行します。
全ての値が0の配列を作成することができました。
指定した通り、要素数は7になっています。

import numpy as np
np.zeros(7)

実行結果:

array([0., 0., 0., 0., 0., 0., 0.])

次に、全ての値が1のは2次元配列を作成してみましょう。
ones関数の引数に、タプルで行数・列数をそれぞれ指定します。
実行します。
全ての値が1の、2次元配列を作成することができました。
指定した通り、2行・4列の2次元配列になっています。

np.ones((2,4))

実行結果:

array([[1., 1., 1., 1.],
       [1., 1., 1., 1.]])

ちなみに、zeros関数とones関数で作成された配列の型は、デフォルトではfroat64です。

np.zeros(7).dtype

実行結果:

dtype('float64')

このデータ型は、引数dtypeで変更することも可能です。

np.zeros(7,dtype=int).dtype

実行結果:

dtype('int32')

なお、任意の数値で配列を作成したい場合には、full関数を使用します。
第一引数に要素数、第二引数に数値を指定します。
数値は、整数・小数いずれも可能です。
実行します。
要素数が5で、それぞれの値が10.5の配列を作成することができました。

np.full(5,10.5)

実行結果:

array([10.5, 10.5, 10.5, 10.5, 10.5])

関連メソッド

【毎日Python】Pythonで配列の形状を変換する方法|numpy.reshape

【こつこつPython】Pythonで要素を指定せずに配列を作成する方法|numpy.empty

【こつこつPython】Pythonで配列をタイル状に繰り返す方法|numpy.tile

【こつこつPython】Pythonで配列の要素を繰り返す方法|numpy.repeat