返回Python中的子列表最大值 - predict_proba的问题？

Question

我有一个Python列表，我试图将每个子列表的最大值放入自己的列表中。我在网上找到了多个解决方案似乎应该为此工作，但我不断收到此错误：TypeError：'list'对象不可调用。这可能是由于嵌套列表首先是如何创建的（来自MultinomialNB模型）。

以下是代码的详细信息。

我创建了一个MultinomialNB模型。

nb_classifier = MultinomialNB()

我已经适应了数据并且正在使用它来预测新数据的分类概率。我将概率输出到列表，这将创建一个嵌套列表：

pred = nb_classifier.predict_proba(stuff_to_assign).tolist()

这似乎工作正常，它似乎创建一个列表列表。 “pred”输出如下所示：

[[0.9679623959303013,
  0.0004919066096350094,
  0.0025707756884416877,
  0.00543455013719153,
  0.009417511656556068,
  0.004419688570339585,
  0.0056471363437299105,
  0.004056035063805071],
 [0.002845247494316791,
  0.000370155483232512,
  0.9749812181697922,
  0.004089452129289453,
  0.007086596337192678,
  0.00332577754892364,
  0.004249421417094243,
  0.003052131420158753]]

我确认这是一个清单：

type(pred)
List

我想要做的是获取每个子列表中的最大值并将其放在自己的列表中。理想输出：

[0.9679623959303013,0.9749812181697922] 

我已经尝试过了：

maxes = [max(p) for p in pred] 

我得到“TypeError：'list'对象不可调用。”关于如何获得理想结果的任何想法？这只是一个嵌套列表问题，还是严格来说，我的嵌套列表是从模型创建的？

Answer 1

它看起来像[ max(val) for val in zip(*pred) ]应该工作。

我希望你的代码返回两个值;第一个列表的最大值和第二个列表的最大值。该错误表明您已在某处向max分配了一个列表。

Answer 2

由于它是嵌套列表，要访问pred中的第一个列表，您需要运行pred[0]。要访问第二个列表，请运行pred[1]

要获得最大值，请分别运行max(pred[0])和max(pred[1])。

你的最终代码会像这样

new_list = []
for i in range(2):
    new_list.append(max(pred[i]))

或者列表理解：

new_list = [max(pred[i]) for i in range(2)]