Advertisement

将 csv 文件的数据进行 保存与导入 操作

阅读量：

在进行网络训练或记录强化学习的回报函数时，采用csv格式能够提升数据的可读性，相较于TXT格式更便于在编辑器中直接查阅。

所依赖的外部库包括：pandas、csv

pandas应用与数据处理

1、保存数据

复制代码 
 import pandas as pd

    
 import random
    
  
    
 # generate data
    
 test_save_data =[]
    
 for episode in range(10):
    
     for step in range(20):
    
     loss = random.random()
    
     test_save_data.append([episode, step, loss])
    
  
    
 # save
    
 # 保存的csv类似Excel表格，header代表每列的名称，header = 0代表没有列名，index为行名
    
 test_dataframe = pd.DataFrame(test_save_data)
    
 test_dataframe.to_csv("test.csv", header = ['episodes','steps','loss'], index=False, s

全部评论 (0)

还没有任何评论哟~

相关文章推荐