使用理解从字符串列表中删除字符

Question

我想知道如何从字符串列表中删除某些字符。 在这种情况下，我正在尝试使用列表理解来删除str类型的numbers 。

numbers = [str(i) for i in range(10)]
imgs_paths = [os.path.join(input_folder, f) for f in os.listdir(input_folder) if f.endswith('.jpg')]
foo_imgs_paths = [[e.replace(c, "") for c in e if c not in numbers] for e in imgs_paths]

上面的代码不起作用，因为它返回完全空的列表。

Answer 1

选项1

如果我正确理解您的问题，则函数可能会比嵌套理解更简化它。

"doj394no.jpg".replace("0","").replace("1","")... # "dojno.jpg"

如果您有要删除的文件列表和字符列表：

files = [...]
numbers = "01234556789"

def remove_chars(original, chars_to_remove):
    for char in chars_to_remove:
        original = original.replace(char, "")
    return original

new_files = [remove_chars(file, numbers) for file in files]

选项 2

如果你真的想使用推导，你可以使用它们来过滤掉字母而不用替换：

numbers = "0123456789"
filename = "log234.txt"
[char for char in filename if char not in numbers] # ["l","o","g",".","t","x","t"]
# To return it to a string:
"".join([char for char in filename if char not in numbers]) # "log.txt"

在你的情况下，它会是这样的：

numbers = [str(i) for i in range(10)]
imgs_paths = [os.path.join(input_folder, f) for f in os.listdir(input_folder) if f.endswith('.jpg')]
foo_imgs_paths = [
    "".join(char for char in img_path if char not in numbers)
    for img_path in img_paths
]

Answer 2

为什么不使用正则表达式？

import re
re.sub(r'\d+', '', 'lo2g4.jpg')
'log.jpg'

Answer 3

只需提供另一种解决方案：

old_str = "S11imone22.jpg"
new_str = old_str.translate(str.maketrans("", "", "0123456789"))
print(new_str) # Simone.jpg

我还是更喜欢re解决方案，速度更快