NumPy transpose()函数
NumPy的transpose()函数可以对数组进行转置操作,即将数组的维度进行调换,让行变成列,列变成行。
import numpy as np a = np.array([[1,2,3],[4,5,6]]) # 输出原数组 print(a) # [[1 2 3] # [4 5 6]] # 转置数组 b = np.transpose(a) print(b) # [[1 4] # [2 5] # [3 6]]
transpose()函数的使用方法有两种:
- np.transpose(arr):这种方法是直接使用NumPy的transpose()函数,可以对数组进行转置操作,返回转置后的数组。
- arr.T:这种方法是直接使用数组的T属性,可以对数组进行转置操作,返回转置后的数组。
transpose()函数还可以接受一个参数,用来指定转置的维度,参数的值可以是一个整数,也可以是一个元组,例如:
# 指定转置的维度 c = np.transpose(a,(1,0)) print(c) # [[1 4] # [2 5] # [3 6]]
NumPy的transpose()函数可以对数组进行转置操作,可以指定转置的维度,使用起来非常方便。