如何从 dtype=object 生成 table 或 DataFrame？

Question

我想展示如何在 table 或 DataFrame 中进行预测。

我试图将 X_test、y_test 和预测 (predictions = model.predict(X_test)) 放入 DataFrame 中，以显示预测的评论是正面的还是负面的.

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"（名称：review，长度：4095，dtype：object）看起来像：

15806 ['tire', 'gauges', 'kind', 'thing', 'makes', '...

541 ['like', 'said', 'title', 'review', 'say', 'pr...

...

变量"y_test"（名称：标签，长度：4095，数据类型：对象）看起来像：

15806 正数

541 正数

...

变量 "predictions" 看起来像：

数组(['positiv', 'positiv', 'positiv', ..., 'positiv', 'positiv', 'positiv'], dtype=object)

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

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 中的格式化输出）来查看整个数据。

如何从 dtype=object 生成 table 或 DataFrame？

How to make a table or DataFrame from dtype=object?

python

dataframe

sentiment-analysis

jupyter-notebook