为什么用pandas.csv后报错说我维度不对?

定义
train_batch_data = csv_data[list(range(0, 4))]

label_data = csv_data[list(range(4,5))]

放数据进神经网络
sess.run(train_step, feed_dict={xs: train_batch_data.loc[i], ys: label_data.loc[i]})

train_batch_data.loc[i]的结果print出来以后的确是某一行的数据,莫非是要我

reform?

建議可以查看其error訊息
通常會告知錯誤原因 可能是row與column對不上