如何将数组存入csv文件中并取出
有些时候我们在运行程序时需要将产生的参数数据存入csv文件中,并且可能在运行程序过程中会存好多次,因此这就需要将数据每次都附加到csv文件每一列的末尾。此处使用的是pandas来处理,以免忘记将其记录下来,并希望能帮到大家。
这里我是将数据存为csv文件的每一列,之前想的是存为每一行,但是没有成功,网上也没查到,因此这里存的是每一列。
#先将数组变为dataframe类型然后存为csv
datefile='write.csv'
a=np.array([5,8,9,123])
df=pd.DataFrame(a)
df.to_csv(datefile)#注意此处数据是以列的形式保存的
#添加新的数据列
b=np.array([6,8,9,34])
df=pd.read_csv(datefile)#读刚才保存的文件
df[1]=b#存第二个数据
df=df.iloc[:,1:4]#去除标签项
df.to_csv(datefile)#存入
#将刚才保存的数据读出为数组
df=pd.read_csv(datefile)#读文件
df=np.asarray(df.iloc[:,2])
print(df)