如何将行列互换
时间: 2024-02-09 16:08:43 浏览: 52
互换数据行列
在NumPy中,可以使用数组的`transpose`方法或者`T`属性来实现行列互换。假设有一个二维数组a,其形状为`(m, n)`,我们可以使用以下两种方法将其行列互换:
1. 使用`transpose`方法:
``` python
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6]])
b = a.transpose()
print(b)
```
这里我们使用`transpose`方法将数组a行列互换,并将结果保存在数组b中。输出结果为:
```
[[1 4]
[2 5]
[3 6]]
```
2. 使用`T`属性:
``` python
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6]])
b = a.T
print(b)
```
这里我们使用数组的`T`属性将数组a行列互换,并将结果保存在数组b中。输出结果为:
```
[[1 4]
[2 5]
[3 6]]
```
以上两种方法等价,可以根据个人喜好选择使用。需要注意的是,这两种方法都只是返回了一个新的行列互换后的数组,并没有修改原始数组。如果需要修改原始数组,可以使用`transpose`方法的`inplace`参数,如下所示:
``` python
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6]])
a.transpose((1, 0)) # 将 a 行列互换并覆盖原数组
print(a)
```
这里我们使用`transpose`方法的`inplace`参数将a行列互换,并覆盖原始数组。输出结果为:
```
[[1 4]
[2 5]
[3 6]]
```
阅读全文