温馨提示×

NumPy数组轴调整的方法是什么

小亿
83
2024-05-13 14:07:34
栏目: 编程语言

在NumPy中,可以使用`numpy.transpose()`函数或`numpy.ndarray.transpose`方法来调整数组的轴

1、使用`numpy.transpose()`函数:

```python

import numpy as np

# 创建一个二维数组

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 调整数组的轴

arr_transposed = np.transpose(arr)

# 输出调整后的数组

print(arr_transposed)

```

输出结果:

```

[[1 4 7]

[2 5 8]

[3 6 9]]

```

2、使用`numpy.ndarray.transpose`方法:

```python

import numpy as np

# 创建一个二维数组

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 调整数组的轴

arr_transposed = arr.transpose()

# 输出调整后的数组

print(arr_transposed)

```

输出结果:

```

[[1 4 7]

[2 5 8]

[3 6 9]]

```

对于多维数组,可以通过指定`axes`参数来调整轴的顺序。例如,对于一个三维数组,可以使用以下代码调整轴顺序:

```python

import numpy as np

# 创建一个三维数组

arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

# 调整数组的轴顺序

arr_transposed = np.transpose(arr, axes=(1, 0, 2))

# 输出调整后的数组

print(arr_transposed)

```

输出结果:

```

[[[1 2]

[5 6]]

[[3 4]

[7 8]]]

```

在这个示例中,我们将三维数组的轴顺序从`(0, 1, 2)`调整为`(1, 0, 2)`。可以看到,数组的形状和元素顺序已经发生了变化。

0