python - 组合来自 for 循环的响应并将它们导出到附加列中以 excel

Question

下面是我的 python 代码。输出正是我正在寻找的，但是我无法弄清楚如何将最后一行 python 代码打印出的多个响应返回到数据框中的新列中。我过去尝试过时得到 None 值。请帮忙!我正在使用 open.ai 和 pandas，并希望导出到 excel 文件。

for question in df['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
    )
    print(response['choices'][0]['text'].strip())

print(response['choices'][0]['text'].strip())

此行为我提供了正确的输出，但我如何连接这些字符串并将它们放入 excel 中的新列中。难倒了!

Answer 1

创建一个返回问题答案的函数，就像您这样做一样:

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)