pandas 合并for循环中的响应,并将它们以附加列的形式导出到Excel

xxb16uws  于 2022-12-09  发布在  其他
关注(0)|答案(1)|浏览(97)

下面是我的python代码。输出正是我所寻找的,但是我不知道如何把python代码的最后一行打印出来的多个响应返回到 Dataframe 中的一个新列。我得到了None值,当我在过去尝试过。请帮助!我正在使用open.ai和pandas,并希望导出到一个excel文件。
第一个
这一行给了我正确的输出,但我如何连接这些字符串,并把它们放在一个新的列在excel。

vmjh9lq9

vmjh9lq91#

创建一个函数,返回问题的答案,就像下面这样:

def openai_completion(question):
    response = openai.Completion.create(
    model="text-davinci-003",
    prompt="Write a detailed, lengthy, and authoratative response to the question: {}".format(question),
    temperature=0.7,
    max_tokens=20,
    top_p=1,
    frequency_penalty=0,
    presence_penalty=0
    )
    return response['choices'][0]['text'].strip()

现在我们需要创建一个名为Answers的新列,该列由apply函数填充,我们将该函数〉apply〈应用于您的Question列:

df["Answer"] = df["Question"].apply(openai_completion)

现在您的DataFrame可以导出为excel。

df.to_excel("MyExcelFile.xlsx", index=False)

相关问题