在行级向熊猫数据框添加列表元素

Question

我有一个返回3个元素列表的函数，我必须在熊猫数据框中输入该列表。函数采用app.$bugsnag.notify(new Error("Bearer tooken is missing in Axios request.")) ，并在列表中推荐3个相似的名称，我必须将其映射。

names

我想要的数据框是

      names             
      a       
      b
      c

函数返回以下输出

      names     recommendations             
      a         d
      a         e
      a         f  
      b         x
      b         y
      b         z
      c         p 
      c         q
      c         r

我正在 def get_recommendations(names, cosine_sim=cosine_sim): idx = indices[names] sim_scores = list(enumerate(cosine_sim[idx])) sim_scores = sorted(sim_scores, key=lambda x: x[1], reverse=True) sim_scores = sim_scores[1:6] movie_indices = [i[0] for i in sim_scores] return list(df['names'].iloc[names_indices]) get_recommendations('a') = ['d', 'e', 'f'] 中执行以下操作，但这不起作用。

Pandas

Answer 1

首先使用apply和function创建新列

然后使用df.explode

df["recommendations"] = df["names"].apply(get_recommendations)

df = df.explode("recommendations")

print(df)