微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

使用列表理解如果嵌套

如何解决使用列表理解如果嵌套

给出一个文件名列表,我们想将所有扩展名为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)

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。