我有一个.dat文件,我想在我脚本中使用它来绘制散点图,从.dat文件中输入数据。为此,我一直在手动将.dat文件转换为.csv,但我发现它并不令人满意。这是我目前使用的。
import pandas as pd import matplotlib.pyplot as plt import numpy as np
filename=raw_input('Enter filename ')
csv = pd.read_csv(filename)
data=csv[['deformation','stress']]
data=data.astype(float)
x=data['deformation']
y=data['stress']
plt.scatter(x,y,s=0.5)
fit=np.polyfit(x,y,15)
p=np.poly1d(fit)
plt.plot(x,p(x),"r--")
plt.show()
一个程序员朋友告诉我,把它转换成JSON使用起来会更方便,我该怎么做呢?
1条答案
按热度按时间cwtwac6a1#
尝试使用numpy读取功能
loadtxt比fromfile更灵活