numpy的ones

numpy的ones

  ones_like的函数原型如下：

s_like(a, dtype=None, order='K', subok=True, shape=None)

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

import numpy as npa = 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]]