一、mat文件操作
1. 加载mat文件
import scipy.io as scio
dataFile = "C:\\pos.mat"
data = scio.loadmat(dataFile) # 加载mat文件
读取的data为字典格式,可以通过type(data)查看
print(type(data))
2. 对读取的数据作切片操作
data_0= data['pos'][:,0]
3. 保存为mat文件
mdic = {"pos":pos}
scio.savemat("pos.mat",mdic)
二、复数数组初始化
必须声明该数组为复数类型,否则后面的运算结果没有虚部
data = np.zeros((n1,n2,n3,dtype='complex_')
且n1、n2、n3参数均必须为int类型
三、数组类型转换
将某个类型的数组转化成另一个类型的数组可以用astype(type)函数
四、数组转置
对于二维数组
A = np.array(range(4)).reshape(2,2)
# 对A转置
print(A.T)
print(np.transpose(A))
对于一维数组,使用以上两种方法后数组维度仍不变,此时可以采用
B = np.array(range(5))
>> B.shape = (5,)
np.array([B])
>> B.shape = (1,5)
五、复共轭转置
matlab中的(.)'表示复共轭转置,python中对应的为:
A.T.conjugate()
六、shape(5, )和(5, 1)的区别
(5, )代表一维数组,数组中有两个元素
(5, 1)代表二维数组,每行有一个元素
如下例:
x = np.array([1, 2, 3, 4, 5])
y = np.array([[1],[2],[3],[4],[5]])
print(x.shape)
>> (5, )
print(y.shape)
>> (5, 1)
将二维数组转换为一维数组:
y = y.flatten()
print(y.shape)
>> (5, )