如何从 dtype=object 创建表或数据帧

Question

我想展示如何在表或数据帧中进行预测。

我尝试将X_test、y_test和预测（预测 = model.predict（X_test））放入数据帧中，以显示哪些评论是正面的或负面的预测。

import pandas as pd
predictions = model.predict(X_test)
df_prediction = pd.DataFrame({
    'X_test': [X_test],
    'y_test': [y_test],
    'prediction': [predictions]
})
df_prediction.head()

变量"X_test"（名称：审阅，长度：4095，dtype：对象）如下所示：

15806 ["轮胎"、"仪表"、"种类"、"东西"、"制造"、"...

541 ["喜欢"、"说"、"标题"、"评论"、"说"、"公关...

。

变量"y_test"（名称：标签，长度：4095，dtype：对象）如下所示：

15806 正

541 正

。

变量"预测"如下所示：

array（['positiv'， 'positiv'， 'positiv'， ...， '

positiv'， 'positiv'，'positiv']， dtype=object）

目前，我得到了一个数据帧，所有数据都在第一行，但我需要一个包含所有行的表。

Answer 1

如果x_test、y_test和predictions是列表，那么你可以这样做：

df_prediction = pd.DataFrame({
    'X_test': x_test,
    'y_test': y_test,
    'prediction': predictions
})

此外，df_prediction.head()将打印数据帧的前 5 行。您可以使用print(df_prediction)或仅使用df_prediction（用于Jupyter Notebook中的格式化输出）来查看整个数据。