如何从python中的字符串中提取由某些字符组成的子字符串？

Question

例如，假设我有一个字符串“beabeefeab”。 我想提取一个仅由 'b' 和 'a' 组成的子字符串，即“babab”。

我通过实现嵌套循环并删除除 'b' 和 'a' 之外的所有字符来应用蛮力

Answer 1

您可以使用简单的列表理解来完成

a = "beabeefeab"
print("".join([i for i in a if (i == 'a' or i =='b')]))

输出：

babab

Answer 2

不是很优雅，但它有效。

a = "beabeefeab"
answer = ""
for char in a:
    if char == "a" or char == "b":
        answer += char

print(answer)

输出

babab

Answer 3

使用set来保留允许的字符，使其更具可扩展性：

s = "beabeefeab"
allowed = set('ab')
print("".join(x for x in s if x in allowed))