numpy数组维度变换

NumPy是一个Python的科学计算库，它提供了一个多维数组对象及一组用于操作数组的函数。在NumPy中，可以通过reshape()函数对数组的维度进行变换。reshape()函数的一个参数为要改变的维度大小，可以是整数或者-1，-1代表自动计算该维度大小。

例如，有一个一维数组a = np.array([1,2,3,4,5,6])，现在需要将其变为一个二维数组，可以使用reshape函数。示例如下：

import numpy as np
a = np.array([1,2,3,4,5,6])
b = a.reshape(2,3)
print(b)

输出结果为：

[[1 2 3]
 [4 5 6]]

如果要将数组的维度变为一维，可以使用flatten()函数。示例如下：

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

输出结果为：

[1 2 3 4 5 6]

同时，还可以使用transpose()函数进行数组的转置操作。示例如下：

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

输出结果为：

[[1 4]
 [2 5]
 [3 6]]

除此之外，还有一些其他的函数可以对数组进行维度变换和转换。需要根据实际需求进行选择使用。