数组到数据框中的列

Question

我已经按照this tutorial 构建了一个有效的分类模型。 我引入了一个 csv，然后将每一行的文本值传递给一个调用分类模型进行预测的函数。该函数返回一个数组，我需要将其放入数据框中的列中。

功能：

def get_top_k_predictions(model,X_test,k):

# get probabilities instead of predicted labels, since we want to collect top 3
np.set_printoptions(suppress=True)
probs = model.predict_proba(X_test)

# GET TOP K PREDICTIONS BY PROB - note these are just index
best_n = np.argsort(probs, axis=1)[:,-k:]

# GET CATEGORY OF PREDICTIONS
preds = [
[(model.classes_[predicted_cat], distribution[predicted_cat])
 for predicted_cat in prediction]
for distribution, prediction in zip(probs, best_n)]


preds=[ item[::-1] for item in preds]
return preds

函数调用：

for index, row in df.iterrows():
category_test_features=category_loaded_transformer.transform(df['Text'].values.astype('U'))
df['PREDICTION'] = get_top_k_predictions(category_loaded_model,category_test_features,9)

这是函数的输出：

[[('Learning Activities', 0.001271131465669718),
  ('Communication', 0.002696299964802842),
  ('Learning Objectives', 0.002774964762863968),
  ('Learning Technology', 0.003557563051027678),
  ('Instructor/TAs', 0.004512712287403168),
  ('General', 0.006675929282872587),
  ('Learning Materials', 0.013051869950436862),
  ('Course Structure', 0.02781481160602757),
  ('Community', 0.9376447176288959)]]

我希望输出最终看起来像 like this。

Answer 1

您的函数返回一个包含元组列表的列表？为什么是双嵌套列表？我能想到的一种方法：

tmp = {}
for index, row in df.iterrows():
    predictions = get_top_k_predictions(...)
    tmp[index] = {
        key: value for key, value in predictions[0]
    }

tmp = pd.DataFrame(tmp).T
df.join(tmp)