首页 > 后端开发 > Python教程 > Python中如何根据字符串输入返回预定义列表?

Python中如何根据字符串输入返回预定义列表?

DDD
发布: 2024-12-18 22:04:15
原创
348 人浏览过

How Can I Return a Predefined List in Python Based on String Input?

根据字符串输入返回预定值

在Python中,我们有时会遇到需要根据提供的字符串返回特定列表的情况。这可以使用多种方法来实现,其中之一涉及利用字典。

使用字典

最直接的方法是定义一个字典,其中键代表字符串输入,对应的值是所需的列表。

get_ext = {'text': ['txt', 'doc'],
            'audio': ['mp3', 'wav'],
            'video': ['mp4', 'mkv']
}
登录后复制

要检索所需的列表,只需访问相应的key:

get_ext['video']
# Output: ['mp4', 'mkv']
登录后复制

在字典中使用函数

如果更喜欢基于函数的解决方案,您可以将字典的 get 方法分配给变量:

get_ext = get_ext.get
登录后复制

此函数将返回指定键的列表,如果键不存在,则返回 None

get_ext('video')
# Output: ['mp4', 'mkv']
登录后复制

使用自定义默认值

要为未知键指定自定义默认值,请考虑使用包装函数:

def get_ext(file_type):
    types = {'text': ['txt', 'doc'],
             'audio': ['mp3', 'wav'],
             'video': ['mp4', 'mkv']
    }

    return types.get(file_type, [])
登录后复制

此函数将对于未知键返回一个空列表。

解决性能问题

对于对于注重性能的应用程序,值得注意的是,每次调用该函数时,都会重新创建包装函数中的类型字典。为了优化这一点,您可以创建一个类并在构造函数中初始化类型字典,确保它只创建一次。

class get_ext(object):
    def __init__(self):
        self.types = {'text': ['txt', 'doc'],
                      'audio': ['mp3', 'wav'],
                      'video': ['mp4', 'mkv']
        }

    def __call__(self, file_type):
        return self.types.get(file_type, [])

get_ext = get_ext()
登录后复制

这种基于类的方法可以轻松修改已识别的文件类型,同时保持性能效率.

以上是Python中如何根据字符串输入返回预定义列表?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板