Linux 拨号vps windows公众号手机端

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

lewis 1年前 (2024-05-07) 阅读数 14 #程序编程
文章标签 numpy

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

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

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方法:

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参数来调整轴的顺序。例如,对于一个三维数组,可以使用以下代码调整轴顺序:

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)。可以看到,数组的形状和元素顺序已经发生了变化。

版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门