使用列表理解如果嵌套

问题描述

给出一个文件名列表,我们想将所有扩展名为hpp的文件重命名为扩展名h。为此,我们要生成一个名为newfilenames的新列表,其中包括新文件名。使用列表理解#

filenames = ["program.c","stdio.hpp","sample.hpp","a.out","math.hpp","hpp.out"]
newfilenames = [filename[:len(filename) - 3] + "h" for filename in filenames if filename.endswith("p")]
print(newfilenames)

输出来自["stdio.h","sample.h","math.h"] 应该是["program.c","stdio.h","math.h","hpp.out"]

解决方法

您可以简单地将.hpp替换为.h

filenames = ["program.c","stdio.hpp","sample.hpp","a.out","math.hpp","hpp.out"]
x=[x.replace(".hpp",".h") for x in filenames]
print (x)

输出:

['program.c','stdio.h','sample.h','a.out','math.h','hpp.out']
,

我认为您需要

filenames = [i[:-3]+"h" if i.split(".")[-1]=="hpp" else i for i in filenames]

print(filenames)

相关问答

错误1:Request method ‘DELETE‘ not supported 错误还原:...
错误1:启动docker镜像时报错:Error response from daemon:...
错误1:private field ‘xxx‘ is never assigned 按Alt...
报错如下,通过源不能下载,最后警告pip需升级版本 Requirem...