Python小坑总结（一）

一、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, )