一日一技:用一个奇技淫巧把字符串转成特定类型

2022-10-27 10:21:00 浏览数 (2)

摄影:产品经理

油炸肥肠

我们有时候可能会需要把一个字符串转换成对应的类型。例如,把'123'转换为int类型的123;或者把'3.14'转成浮点数3.14

前提条件是不能使用eval或者exec

这是一个非常简单的功能,常规做法直接使用if判断就可以了:

代码语言:javascript复制
def convert(data, target_type):
    if target_type == 'int':
        return int(data)
    elif target_type == 'float':
        return float(data)
    ...

有些同学觉得写if判断麻烦,也可能会用字典来处理:

代码语言:javascript复制
def convert(data, target_type):
    type_map = {
        'int': int,
        'float': float,
         ...
    }
    return type_map.get(target_type, str)(data)

但是这样做有个弊端,就是你需要把能够转换的格式都列出来。如果新增了一个格式,你还需要改动代码增加一个elif分支或者在字典新增一个键值对。

那么有没有什么办法,能够在不改动代码的情况下,完成转换呢?

一开始我也想不到什么好办法。直到今天看Scrapy源代码[1]的时候,发现了一段代码:

这段代码中的type(custom)(convert(c) for c in custom)看起来很奇怪,但是只要解构一下,就会变得很简单。今天我们要解决的问题,就是这一行代码的一部分。

先来看前半截的写法:type(custom)()。怎么type后面有两个括号?我们知道type(xxx)是返回xxx这个数据的类型:

有些人以为,type(xxx)返回的是一个字符串。但实际上,它返回的就是类型本身:

既然我们可以使用int('123')把字符串转换为int,那么我们也可以使用type(1)('123'),把字符串'123'转换为int。

所以,今天我们的这个问题,解法就很简单了:

代码语言:javascript复制
def convert(data, sample):
    return type(sample)(data)

调用函数的时候,传入两个参数,第一个是需要转换的字符串,第二个参数,是任意目标类型的数据。运行效果如下图所示:

本来文章到这里就结束了。但考虑到有同学可能不明白上面代码type(custom)(convert(c) for c in custom)中的convert(c) for c in custom看起来像是列表推导式,却少了方括号,我再解释一下。

例如当你一个只含有数字的列表,你要把每一个数字乘以2,然后再传到函数里面,你一般会这样写:

代码语言:javascript复制
def get_one_ele(data_list: List):
    print('具体的执行代码')

a = [1, 2, 3]
get_one_ele([x * 2 for x in a])

但是如果函数只有这一个参数时,你可以使用生成器推导式省略外层的圆括号,简写为:get_one_ele(x * 2 for x in a)。所以上面的代码type(custom)(convert(c) for c in custom)等效为:

代码语言:javascript复制
a = (convert(c) for c in custom)
type(custom)(a)

参考资料

[1]

Scrapy源代码: https://github.com/scrapy/scrapy/blob/master/scrapy/utils/conf.py#L50

0 人点赞