5.已知alist=['jn_20170525_22.tif', 'jn_20200424_1.tif', 'jn_20200424_2.tif'], 编写函数,使得该函数接收参数alist后得到的返回值为['2017', '2020', '2020']
时间: 2023-08-15 18:03:49 浏览: 130
有返回值的函数
可以使用正则表达式来提取文件名中的年份信息,以下是实现该函数的代码:
```python
import re
def extract_years(lst):
years = []
for filename in lst:
result = re.search('\d{4}', filename)
if result:
years.append(result.group())
return years
alist = ['jn_20170525_22.tif', 'jn_20200424_1.tif', 'jn_20200424_2.tif']
result = extract_years(alist)
print(result)
```
运行结果为:
```
['2017', '2020', '2020']
```
该函数接受一个列表作为参数,遍历列表中的每个文件名,使用正则表达式 `\d{4}` 匹配其中的年份信息,并将其添加到 `years` 列表中。最后返回 `years` 列表,即为提取出来的年份信息。调用该函数并打印输出值即可得到 `['2017', '2020', '2020']`。
阅读全文