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

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

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

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

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

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

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

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

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

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

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

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

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

所以，今天我們的這個問題，解法就很簡單了：

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

調(diào)用函數(shù)的時候，傳入兩個參數(shù)，第一個是需要轉(zhuǎn)換的字符串，第二個參數(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)式，卻少了方括號，我再解釋一下。

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

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

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

但是如果函數(shù)只有這一個參數(shù)時，你可以使用生成器推導(dǎo)式省略外層的圓括號，簡寫為：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)