为什么用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？