将处理后的文本存储在熊猫数据框中

Question

我使用 gensim 在 Python 中进行文本摘要。我希望我的汇总输出存储在同一数据框中的不同列中。

我用过这段代码：

for n, row in df_data_1.iterrows():
        text=df_data_1['Event Description (SAP)']
        print(text)
        *df_data_1['Summary']=summarize(text)*
print(df_data_1['Summary'])

此代码的第 4 行出现错误，其中指出：TypeError: expected string or bytes-like object。

如何将处理后的文本存储在 pandas 数据框中

Answer 1

如果它不是字符串或类似字节的，那是什么？您可以检查汇总函数的类型并从那里继续。

test_text = df_data_1['Event Description (SAP)'].iloc[0]
print(type(summarize(test_text))

另一句话：通常您希望避免循环遍历数据帧（请参阅discussion）。如果要将函数应用于整个列，请使用df.apply()，如下所示：

df_data1[‘Summary’] = df_data1['Event Description (SAP)'].apply(lambda x: summarize(x))