我有一系列概率:
proba = :[[0.254 0.556 0.025] [0.898 0567 .112]]
希望每个人的最大值为[[0.556] [0.898]]
我怎么做? 试过2种方法:
1. max(sublist) for sublist in proba
2. proba = map(max,proba)
并收到错误"TypeError: 'numpy.float32' object is not iterable"
有什么建议吗?
答案 0 :(得分:1)
我可以注意到代码中的一些问题, 首先,您的数据列表格式不正确, 逗号丢失,等号右边有一个额外的':'
proba = [[0.254, 0.556, 0.025], [0.898, 0.567, .112]]
然后你可以得到这样的答案:
max_ = [max(i) for i in proba]