替换字符串列表中的整个字符串

Question

我有一个令牌列表。一些以@标志开头。我想将所有这些字符串更改为通用@user。我试过这个：

>>> words = ['@john', 'nina', 'michael', '@bebeto']
>>> words = [w.replace(w, '@user') for w in words if w.startswith('@')]
>>> words
['@user', '@user']
>>> 

我在这里弄错了什么？

Answer 1

首先，您可以简化列表理解的第一部分。这是等效的，不做不必要的替换：

words = ['@user' for w in words if w.startswith('@')]

在列表推导中，最后的if子句决定包含的内容。所以if基本上说，只保留以'@'开头的元素。但是你想要保留所有元素。

相反，您可以使用条件表达式来决定是否获取“@user”或原始单词：

words = ['@user' if w.startswith('@') else w for w in words]

Answer 2

您的列表理解会导致不需要的输出，只需更改即可

[w.replace(w, '@user') for w in words if w.startswith('@')]

至

[w.replace(w, '@user') if w.startswith('@') else w for w in words]

Answer 3

你可以试试这个：

words = ['@john', 'nina', 'michael', '@bebeto']
new_words = ['@user' if i.startswith('@') else i for i in words]

输出：

['@user', 'nina', 'michael', '@user']