Liu/IO.py

35 lines
977 B
Python
Raw Normal View History

2023-11-27 15:56:15 +08:00
#====================================================================
# 保存和输入matlat格式数据并画图显示
#
#====================================================================
import scipy.io as sio
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
# 创建4个变量并赋值
#sio.savemat('data/testpython.mat', {'a': 1, 'b': 5, 'c': 3, 'd': 4})
# 创建了一个变量x并赋予一个矩阵
#sio.savemat('data/testpython2.mat', {'x': [[1, 3, 5, 4], [5, 3, 2, 8]]})
data = sio.loadmat('data/testpython.mat')
data2 = sio.loadmat('data/testpython2.mat')
x=np.zeros([1,4])
x[0][0] = data['a']
x[0][1] = data['b']
x[0][2] = data['c']
x[0][3] = data['d']
y=data2['x']
print(x,x.dtype,y[0],y[0].dtype)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
xs = x
ys = y[0]
zs = y[1]
ax.scatter(xs, ys, zs, c='b', marker='o')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
plt.show()
plt.close()