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