在numpy中,可以使用函数numpy.tile()
来为第i轴上的每个第i个向量分配相同的值。numpy.tile()
函数可以将一个数组沿指定轴向堆叠多次,从而实现复制的效果。
具体步骤如下:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
shape = arr.shape
result = np.tile(5, shape)
5
表示要分配的相同值,shape
表示原始数组的形状。print(result)
完整代码示例:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
shape = arr.shape
result = np.tile(5, shape)
print(result)
输出结果为:
[[5 5 5]
[5 5 5]
[5 5 5]]
这样,我们就成功为第i轴上的每个第i个向量分配了相同的值。
领取专属 10元无门槛券
手把手带您无忧上云