使用python替换列表中的字符串，并使用其他变量保存它

Question

我试图用列表中的另一个字符串替换一个字符串，但失败了，我不知道为什么

例如，我有一个列表：

predicts = 
[['__label__1'],
 ['__label__0'],
 ['__label__1'],
 ['__label__1'],
 ['__label__0'],
 ['__label__1']]

我想用“OK”替换\uuuu label\uu 1，用“NOT OK”替换\uuuu label\uu 0，并使用以下命令将其保存在不同的变量中：

pred_label = []
for i in predicts:
  if i == '__label__1':
    pred_label.append("OK")
  else:
    pred_label.append("NOT OK")

但它没能取代任何一个

Answer 1

预测变量有两个维度，请尝试：

pred_label = []
for i in predicts:
    if i[0] == '__label__1':
        pred_label.append("OK")
    else:
        pred_label.append("NOT OK")
print(pred_label)

Answer 2

您可以使用字典映射您希望进行的替换：

substituteDict = {'__label__1': 'OK', '__label__0': 'NOT OK'}

replaced_list = []
for i in predicts:
    replaced_list.append(substituteDict.get(i[0]))

print(replaced_list)