Python 数组扩维代码示例:将二维数组扩展为三维
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 将二维数组的第一维扩展为一个新的维度,从而得到了一个三维数组。
你可以根据需要将这个代码嵌入到你的程序中,来实现数组扩维的操作。
原文地址: https://www.cveoy.top/t/topic/lmk0 著作权归作者所有。请勿转载和采集!