numpy的ones_like函数

时间:2024-11-18 08:09:54

返回一个用1填充的跟输入 形状和类型 一致的数组。

>>> x = (6)
>>> x = ((2, 3))
>>> x
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.ones_like(x)
array([[1, 1, 1],
       [1, 1, 1]])
>>>
>>> y = (3, dtype=)
>>> y
array([ 0.,  1.,  2.])
>>> np.ones_like(y)
array([ 1.,  1.,  1.])

同理,zeros_like返回一个用0填充的跟输入数组 形状和类型一样的数组。