已知alist=['jn_20170525_22.tif', 'jn_20200424_1.tif', 'jn_20200424_2.tif'], 编写函数,使得该函数接收参数alist后得到的返回值为['2017', '2020', '2020']
时间: 2023-08-06 17:16:48 浏览: 152
有返回值的函数
可以通过字符串的切片操作和列表推导式来实现:
```python
def get_years(alist):
years = [filename.split("_")[1][:4] for filename in alist]
return years
```
在这个函数中,我们首先使用了列表推导式,循环遍历了`alist`中的每一个元素`filename`,然后使用字符串的`split()`方法将文件名拆分成一个列表,其中第二个元素就是日期,我们再使用切片操作获取该日期的前四个字符(即年份),最后将所有年份组成的列表返回即可。
使用该函数可以得到如下结果:
```python
>>> alist = ['jn_20170525_22.tif', 'jn_20200424_1.tif', 'jn_20200424_2.tif']
>>> get_years(alist)
['2017', '2020', '2020']
```
阅读全文