python数组增加维度

可以使用numpy库中的reshape方法将一维或二维数组增加维度。例如，将一维数组a增加为二维数组b：

import numpy as np

a = np.array([1, 2, 3, 4])
b = a.reshape(2, 2)
print(b)

输出结果为：

[[1 2]
 [3 4]]

将二维数组c增加为三维数组d：

c = np.array([[1, 2], [3, 4]])
d = c.reshape(1, 2, 2)
print(d)

输出结果为：

[[[1 2]
  [3 4]]]

reshape方法的参数是一个元组，元组中的每个值表示对应维度的大小。如果元组中有一个维度的大小为-1，则该维度的大小将根据数组总大小自动计算。