ValueError：索引数据必须是一维的

Question

我正在尝试使用 RANDOM FOREST actual 制作数据框。 预测值和来自X_test的所有信息。

这是我的代码：

def makeDataframe(self):
    self.df.index = self.X_test.reset_index()
    self.df['type'] = self.y_test
    self.df['prediction'] = self.y_pred
    self.df['ax'] = self.X_test['ax']
    self.df['ay'] = self.X_test['ay']
    self.df['az'] = self.X_test['az']
    print(self.df.head(10))

我想做一个这样的数据框：

type  prediction. ax ay az

但我遇到了这个错误：

ValueError: Index data must be 1-dimensional

解决此错误的任何想法。 此外，我想确认我的预测索引和 x_test 数据集索引是否匹配。

我能得到一些帮助吗？

谢谢

Answer 1

您收到上述错误是因为 reset_index() 将返回一个 DataFrame 而不是一维数据，您可以将其替换为self.X_test.index.values或要创建一个 DataFrame，您可以使用这样的 Dictionary：

def makeDataframe(self):
    data = {}
    data.update({'index': self.X.index.values})
    data.update({'type': self.y_test.values})
    data.update({'prediction': self.y_pred})
    data.update({'ax': self.X_test['ax']})
    data.update({'ay': self.X_test['ay']})
    data.update({'az': self.X_test['az']}) 
    self.df = pd.DataFrame(data)
    print(self.df.head(10))