我們有時(shí)候可能會(huì)需要把一個(gè)字符串轉(zhuǎn)換成對(duì)應(yīng)的類型。例如，把'123'?轉(zhuǎn)換為int?類型的123?；或者把'3.14'?轉(zhuǎn)成浮點(diǎn)數(shù)3.14。

前提條件是不能使用eval?或者exec。

這是一個(gè)非常簡(jiǎn)單的功能，常規(guī)做法直接使用if判斷就可以了：

def convert(data, target_type):
    if target_type == 'int':
        return int(data)
    elif target_type == 'float':
        return float(data)
    ...

有些同學(xué)覺得寫if判斷麻煩，也可能會(huì)用字典來處理：

def convert(data, target_type):
    type_map = {
        'int': int,
        'float': float,
         ...
    }
    return type_map.get(target_type, str)(data)

但是這樣做有個(gè)弊端，就是你需要把能夠轉(zhuǎn)換的格式都列出來。如果新增了一個(gè)格式，你還需要改動(dòng)代碼增加一個(gè)elif分支或者在字典新增一個(gè)鍵值對(duì)。

那么有沒有什么辦法，能夠在不改動(dòng)代碼的情況下，完成轉(zhuǎn)換呢？

一開始我也想不到什么好辦法。直到今天看??Scrapy源代碼??的時(shí)候，發(fā)現(xiàn)了一段代碼：

這段代碼中的type(custom)(convert(c) for c in custom)看起來很奇怪，但是只要解構(gòu)一下，就會(huì)變得很簡(jiǎn)單。今天我們要解決的問題，就是這一行代碼的一部分。

先來看前半截的寫法：type(custom)()?。怎么type?后面有兩個(gè)括號(hào)？我們知道type(xxx)?是返回xxx這個(gè)數(shù)據(jù)的類型：

有些人以為，type(xxx)返回的是一個(gè)字符串。但實(shí)際上，它返回的就是類型本身：

既然我們可以使用int('123')?把字符串轉(zhuǎn)換為int，那么我們也可以使用type(1)('123')?，把字符串'123'轉(zhuǎn)換為int。

所以，今天我們的這個(gè)問題，解法就很簡(jiǎn)單了：

def convert(data, sample):
    return type(sample)(data)

調(diào)用函數(shù)的時(shí)候，傳入兩個(gè)參數(shù)，第一個(gè)是需要轉(zhuǎn)換的字符串，第二個(gè)參數(shù)，是任意目標(biāo)類型的數(shù)據(jù)。運(yùn)行效果如下圖所示：

本來文章到這里就結(jié)束了。但考慮到有同學(xué)可能不明白上面代碼type(custom)(convert(c) for c in custom)?中的convert(c) for c in custom看起來像是列表推導(dǎo)式，卻少了方括號(hào)，我再解釋一下。

例如當(dāng)你一個(gè)只含有數(shù)字的列表，你要把每一個(gè)數(shù)字乘以2，然后再傳到函數(shù)里面，你一般會(huì)這樣寫：

def get_one_ele(data_list: List):
    print('具體的執(zhí)行代碼')

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

但是如果函數(shù)只有這一個(gè)參數(shù)時(shí)，你可以使用生成器推導(dǎo)式省略外層的圓括號(hào)，簡(jiǎn)寫為：get_one_ele(x * 2 for x in a)?。所以上面的代碼type(custom)(convert(c) for c in custom)等效為：

a = (convert(c) for c in custom)
type(custom)(a)