numpyのones_like関数
440 ワード
1で埋め込まれた入力シェイプとタイプが一致する配列を返します.
同じ理屈でzeros_likeは、入力配列の形状とタイプと同じ0で埋め込まれた配列を返します.
>>> x = np.arange(6)
>>> x = x.reshape((2, 3))
>>> x
array([[0, 1, 2],
[3, 4, 5]])
>>> np.ones_like(x)
array([[1, 1, 1],
[1, 1, 1]])
>>>
>>> y = np.arange(3, dtype=np.float)
>>> y
array([ 0., 1., 2.])
>>> np.ones_like(y)
array([ 1., 1., 1.])
同じ理屈でzeros_likeは、入力配列の形状とタイプと同じ0で埋め込まれた配列を返します.