例2.5 在列表推导式中使用if过滤不符合条件的元素

例2.5 在列表推导式中使用if过滤不符合条件的元素

2.5.1

import os
fn = [filename for filename in os.listfir('D:\Programs\Python\Python37') if filename.endswith(('.exe','.py'))]
print(fn)

print("学号：3004")

2.5.2

from numpy.random import randint
import numpy as np
a=randint(10,20,16)
ma=max(a)
ind1=[index for index, value in enumerate(a) if value == ma]
ind2=np.where(a==ma)
print(ind1);print(ind2[0])

print("学号：3004")