Python 数组扩维代码示例:将二维数组扩展为三维

在 Python 中,可以使用 NumPy 库来实现数组的扩维。下面的代码演示了如何将一个二维数组扩展为三维数组。

import numpy as np

# 创建一个二维数组
arr2d = np.array([[1, 2], [3, 4]])

# 扩展为三维数组
arr3d = arr2d[np.newaxis, :, :]

print('原数组:')
print(arr2d)

print('扩展后的数组:')
print(arr3d)

在这个例子中,我们使用 np.newaxis 将二维数组的第一维扩展为一个新的维度,从而得到了一个三维数组。

你可以根据需要将这个代码嵌入到你的程序中,来实现数组扩维的操作。

Python 数组扩维代码示例:将二维数组扩展为三维

原文地址: https://www.cveoy.top/t/topic/lmk0 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录