如何为列表中的每个项目移动字符串的一部分

Question

I have made a program in Python which should, for every item in the list, move a certain substring "{Organization}" to the end of the item.

The list: example_list = ['Wall{Organizationmart', 'is', 'a', 'big', 'company']

This is the code I made

output = []
word = '{Organization'
for i in example_list:
    output.append(i.replace(word, "") + str(word) + "}")
print(output)

The expected output is: ['Wallmart{Organization}', 'is', 'a', 'big', 'company']

However, this is the output:

['Wallmart{Organization}', 'is{Organization}', 'a{Organization}', 'big{Organization}', 'company{Organization}']

Any help would be appreciated. Thank you very much.

Answer 1

You forgot to check if organization is in each string. Slight modification to your code:

output = []
word = '{Organization'
for i in example_list:
    if word in i:
        output.append(i.replace(word, "") + str(word) + "}")
    else:
        output.append(i)

print(output)

Output:

['Wallmart{Organization}', 'is', 'a', 'big', 'company']

---

Same thing with list comprehension:

output = [i.replace(word, "") + str(word) + "}" if word in i else i for i in example_list]

Answer 2

You must check if the word is in the list element and decide what to print according to that. The solution to your problem is:

example_list = ['Wall{Organizationmart', 'is', 'a', 'big', 'company']

output = []
word = '{Organization'
for i in example_list:
    if word in i:
        output.append(i.replace(word, "") + str(word) + "}")
    else:
        output.append(i)
print(output)