我想读取y-predict，让它每一行若第一个数大于第二个数就写成1，0，否则0，1的代码怎么写

Question

我想读取y-predict，让它每一行若第一个数大于第二个数就写成［1，0］，否则［0，1］，我找了半天都找不到正确的代码怎么写

Answer 1

以下内容由CHATGPT及阿里嘎多学长共同生成、有用望采纳：

---

根据你提供的截图内容，看起来你已经有了一个基本的Python脚本，但是有一些语法错误和逻辑问题。下面是修正后的代码示例，它将实现你所描述的功能：

# 假设y_predict是一个二维数组，每个元素是一个包含两个概率值的列表
y_predict = [
    [0.214191305, 0.780808082],
    [0.755959570, 0.244040474],
    [0.882864237, 0.117135726],
    # ... 其他行
]

# 初始化一个空列表来存储转换后的结果
predict_1 = []

# 遍历y_predict中的每个元素
for row in y_predict:
    # 检查第一个元素是否大于第二个元素
    if row[0] > row[1]:
        # 如果是，则添加[1, 0]
        predict_1.append([1, 0])
    else:
        # 否则，添加[0, 1]
        predict_1.append([0, 1])

# 打印转换后的数据
print(predict_1)

请注意，上面的代码假设y_predict是一个列表，其中每个元素都是一个包含两个浮点数的列表。你需要根据你的具体情况调整这个假设。

关于参考资料，我可以推荐一些在线资源，你可以在这些网站上查找Python编程相关的教程和文档：

• Python官方文档
• W3Schools Python教程
• 菜鸟教程 - Python3教程

这些资源可以帮助你更好地理解Python编程，特别是列表操作和条件语句。