# 过滤掉和值超过6或小于6的预测值 y_pred_filtered = y_pred_prob[(y_pred_prob.iloc[:, :6].sum(axis=1) == 6)]
时间: 2024-02-13 13:07:14 浏览: 77
Code.rar_PRED-163_matlab pred_社交网络_社交网络分析 链路预测_链路预测
5星 · 资源好评率100%
好的,你的第一个问题是如何过滤掉和值超过6或小于6的预测值?回答如下:
可以使用 Pandas 库中的 iloc 和 sum 方法实现。假设 y_pred_prob 是一个 DataFrame,每一行代表一个预测结果,每一列代表一个可能出现的点数,可以通过 y_pred_prob.iloc[:, :6].sum(axis=1) 计算每一行前六列的和,然后使用 ==6 条件筛选出和为 6 的行,最后使用这个条件对 y_pred_prob 进行索引,得到 y_pred_filtered。
具体代码如下:
```
y_pred_filtered = y_pred_prob[(y_pred_prob.iloc[:, :6].sum(axis=1) == 6)]
```
阅读全文