数组扩张一个维度 python

Posted 2023-03-08 K同学啊

要将一个数组在 Python 中扩展一个维度，可以使用 NumPy 库中的函数 numpy.newaxis。这个函数会将原始数组的形状增加一个长度为 1 的新维度。

下面是一个例子，将一个一维数组扩展成一个二维数组：

pythonimport numpy as np

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

# 将数组扩展成一个二维数组
b = a[np.newaxis, :]

print(a.shape)  # 输出 (4,)
print(b.shape)  # 输出 (1, 4)

在这个例子中，我们创建了一个一维数组 a，然后使用 np.newaxis 函数将其扩展成了一个二维数组 b。在扩展后的数组 b 中，第一个维度的长度为 1，第二个维度的长度为原始数组的长度 4。