numpy的ones_like函数

  ones_like的函数原型如下：

numpy.ones_like(a, dtype=None, order='K', subok=True, shape=None)

函数返回一个大小和类型与a相等，且元素全为1的数组。

import numpy as np

a = np.arange(6).reshape((2, 3))
print("a =", a)
b = np.ones_like(a)
print("b =", b)

执行结果：

a = [[0 1 2]
     [3 4 5]]
b = [[1 1 1]
     [1 1 1]]