Python中的 numpy.ones_like()

在Python的科学计算中，常常需要对一个数组进行操作并生成一个形状相同的数组。numpy.ones_like()可以生成一个和输入数组大小和类型相同的由1组成的数组。

用法

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

参数

• a：输入数组。
• dtype：输出数组的数据类型，默认为输入数组类型。
• order：内存中数组元素的顺序，默认为'C'。
• subok：输出数组是否与输入数组具有相同的子类类型，默认为True。
• shape：指定输出数组维度的元组，如果为None则与输入数组大小相同。

返回值

一个大小和类型与输入数组相同的由1组成的数组。

示例

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])
b = np.ones_like(a)

print(b)

输出结果为：

array([[1, 1, 1],
       [1, 1, 1]])

总结

numpy.ones_like()函数生成了一个和输入数组大小及类型相同的由1组成的数组，常用于科学计算中的数组操作。