文章目录
列表推导式
列表推导式1:(list comprehension)
folders = [folder for folder in os.listdir() if os.path.isdir(folder)]
os.listdir()
方法返回指定目录中的文件和文件夹的名称列表。如果在调用该方法时没有传递地址参数,它将返回当前工作目录(即脚本所在的目录)下的所有文件和文件夹的名称。
列表推导式[folder for folder in os.listdir() if os.path.isdir(folder)]
遍历os.listdir()
返回的列表,检查每个条目:
for folder in os.listdir()
表示遍历os.listdir()
返回的列表,
而if os.path.isdir(folder)
表示只有当当前遍历的条目是文件夹时才会包含在最终的folders列表中。
因此,最终folders列表中包含的是当前工作目录下所有文件夹的名称。
列表推导式2:
a = [1 if p >= 0.5 else 0 for p in prob] # if p >= 0.5,向a输出1;否则,向a输出0
correct = [1 if a == b else 0 for (a, b) in zip(X, Y)] # 使用zip()将多个序列中的元素进行配对,然后成对迭代
enumerate()
enumerate():
for i, segment in enumerate(list_example):
enumerate()
同时获取列表元素的索引值(i)和相应的元素内容(segment),返回结果以元组存储为一个列表。
list_example = ["a", "b", "c"]
print(list(enumerate(list_example)))
[(0, 'a'), (1, 'b'), (2, 'c')]