Replace string in a list using python and save it with different variable

Question

I am trying to replace a string with another string in a list, but failed and I don't know why.

For example, I have a list:

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

i want to replace __label__1 with "OK" and __label__0 with "NOT OK" and save it in different variable using :

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

but it failed to replace any of it

There is both a predict and predicts variable in your code, choose one or predicts is considered as empty in the second part of your code. — vgalin
– vgalin, Commented Aug 15, 2020 at 16:16
using list comprehension predicts = [["OK"] if item[0] is "__label__1" else ["NOT OKE"] for item in predicts] — Kamal Lama
– Kamal Lama, Commented Aug 15, 2020 at 16:35
As @kevin-mayo points out, you have a list of lists, not a list of strings. Is this a requirement of what you are trying to accomplish? — jstevenco
– jstevenco, Commented Aug 15, 2020 at 19:02

jstevenco · Accepted Answer · 2020-08-15 19:11:39Z

2

The predicts variable has two dimensions, try:

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

edited Aug 15, 2020 at 19:11

jstevenco

2,9532 gold badges26 silver badges36 bronze badges

answered Aug 15, 2020 at 16:27

Kevin Mayo

1,1591 gold badge8 silver badges19 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

jstevenco · Accepted Answer · 2020-08-15 19:21:48Z

0

You could use a dictionary to map the replacement you wish to make:

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

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

print(replaced_list)

edited Aug 15, 2020 at 19:21

jstevenco

2,9532 gold badges26 silver badges36 bronze badges

answered Aug 15, 2020 at 16:54

Krishna Singhal

6616 silver badges9 bronze badges

Collectives™ on Stack Overflow

Replace string in a list using python and save it with different variable

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related