Pythonnumpy多维数组

Question

如何让我生成一个随机多维数组中每个数组的最后两个数的值恒为零呢？

Answer 1

引自免费微信小程序：皆我百晓生

这是一个很好的问题，我们可以使用numpy库中的函数来生成满足条件的多维数组。以下是一个可能的解决方案：

首先，我们需要导入numpy库。然后，我们可以创建一个多维数组，并使用切片操作在最后两个位置填充零。这是一个示例代码：

import numpy as np

def generate_array_with_last_two_zeros(shape, value=None):
    # 生成一个指定形状的全值数组，值为value或随机值（如果未指定value）
    arr = np.full(shape, value if value is not None else np.random.rand())
    # 将最后两个元素设置为零
    arr[..., -2:] = 0
    return arr

# 使用示例：生成一个形状为(3, 4)的二维数组，其中每个数组的最后两个数恒为零。
array = generate_array_with_last_zeros((3, 4))
print(array)

这个函数generate_array_with_last_zeros接收一个形状参数（一个表示数组维度的元组），以及一个可选的初始值参数。函数首先生成一个全值数组（如果未指定初始值，则为随机值），然后将每个子数组的最后两个元素设置为零。这样我们就得到了一个多维数组，其中每个数组的最后两个数的值恒为零。