读取多个.csv文件以在Tensorflow中进行训练和测试

Question

我只是使用tensorflow的初学者。我用python编写了一个代码，该代码正在读取一个.csv文件，并将其拆分为训练和测试。这是我的操作方式：

diode_df=pd.read_csv("./fun.csv", sep=",",names=['voltage','current'])

但是，现在，我大约有1000个.csv文件，每个文件都有功能和标签列。我想读取说1000个文件中的900个用于培训，其余100个文件用于测试。有人可以帮忙吗？

Answer 1

import os
import pandas as pd

files = os.path.listdir('./')
train = pd.DataFrame()
test = pd.DataFrame()
for file in files:
  df = pd.read_csv(file, sep=",",names=['voltage','current'])
  if i < 900:
    train.append(df)
  else:
     test.append(df)