摘要：上個月，學習群里的同學問了個題目，大意可理解為列表降維，例子如下想得到結(jié)果原始數(shù)據(jù)是一個二維列表，目的是獲取該列表中所有元素的具體值。不經(jīng)意間，函數(shù)的注意事項，竟把其它的進階內(nèi)容都聯(lián)系起來了。小小的函數(shù)，竟成為學習之路上的一個樞紐。

上個月，學習群里的 S 同學問了個題目，大意可理解為列表降維 ，例子如下：

oldlist = [[1, 2, 3], [4, 5]]

# 想得到結(jié)果：
newlist = [1, 2, 3, 4, 5]

原始數(shù)據(jù)是一個二維列表，目的是獲取該列表中所有元素的具體值。從抽象一點的角度來理解，也可看作是列表解壓或者列表降維。

這個問題并不難，但是，怎么寫才比較優(yōu)雅呢？

# 方法一，粗暴拼接法：
newlist = oldlist[0] + oldlist[1]

這種方法簡單粗暴，需要拼接什么內(nèi)容，就取出來直接拼接。然而，如果原列表有很多子列表，則這個方法就會變得繁瑣了。

我們把原問題升級一下：一個二維列表包含 n 個一維列表元素，如何優(yōu)雅地把這些子列表拼成一個新的一維列表？

方法一的做法需要寫 n 個對象，以及 n - 1 次拼接操作。當然不可行。下面看看方法二：

# 方法二，列表推導式：
newlist = [i for j in range(len(oldlist)) for i in oldlist[j]]

這個表達式中出現(xiàn)了兩個 for 語句，在第一個 for 語句中，我們先取出原列表的長度，然后構(gòu)造 range 對象，此時 j 的取值范圍是 [0, n-1] 的閉區(qū)間。

在第二個 for 語句中，oldlist[j] 指的正是原列表的第 j 個子列表，for i in oldlist[j] 則會遍歷取出 j 子列表的元素，由于 j 取值的區(qū)間正對應于原列表的全部索引值，所以，最終達到解題目的。

這種方法足夠優(yōu)雅了，而且理解也并不難。

然而，我們是否就能滿足于此了呢？有沒有其它奇技淫巧，哦不，是其它高級方法呢？F 同學貢獻了一個思路：

# 方法三，巧用sum：
newlist = sum(oldlist,[])

說實話，這個方法令我大感意外！sum() 函數(shù)不是用于求和的么？怎么竟然有此用法？

這個寫法利用了什么原理呢？由于我開始時不知道 sum() 函數(shù)可以接收兩個參數(shù)，不清楚它們是怎么用于計算的，所以一度很困惑。但是，當我知道 sum() 的完整用法時，我恍然大悟。

接下來也不賣關子了，直接揭曉吧。

語法： sum(iterable[, start]) ，sum() 函數(shù)的第一個參數(shù)是可迭代對象，如列表、元組或集合等，第二個參數(shù)是起始值，默認為 0 。其用途是以 start 值為基礎，再與可迭代對象的所有元素相“加”。

在上例中，執(zhí)行效果是 oldlist 中的子列表逐一與第二個參數(shù)相加，而列表的加法相當于 extend 操作，所以最終結(jié)果是由 [] 擴充成的列表。

這里有兩個關鍵點：sum() 函數(shù)允許帶兩個參數(shù)，且第二個參數(shù)才是起點。 可能 sum() 函數(shù)用于數(shù)值求和比較多，然而用于作列表的求和，就有奇效。它比列表推導式更加優(yōu)雅簡潔！

至此，前面的升級版問題就得到了很好的回答。簡單回顧一下，s 同學最初的問題可以用三種方法實現(xiàn)，第一種方法中規(guī)中矩，第二種方法正道進階，而第三種方法旁門左道（沒有貶義，只是說它出人意料，卻效果奇佳）。

這道并不算難的問題，在眾人的討論與分享后，竟還引出了很有價值的學習內(nèi)容。前不久，同樣是群內(nèi)的一個問題，也產(chǎn)生了同樣的學習效果，詳見《Python進階：如何將字符串常量轉(zhuǎn)為變量？》。

我從中得到了一個啟示：應該多角度地思考問題，設法尋求更優(yōu)解，同時，基礎知識應掌握牢固，并靈活貫通起來。

學無止境，這里我還想再開拓一下思路，看看能發(fā)現(xiàn)些什么。

1、如果原列表的元素除了列表，還有其它類型的元素，怎么把同類的元素歸并在一起呢？

2、如果是一個三維或更高維的列表，怎么更好地把它們壓縮成一維列表呢？

3、sum() 函數(shù)還有什么知識要點呢？

前兩個問題增加了復雜度，解決起來似乎沒有“靈丹妙藥”了，只能用笨方法分別拆解，逐一解壓。

第三個思考題是關于 sum() 函數(shù)本身的用法，我們看看官方文檔是怎么說的：

The iterable’s items are normally numbers, and the start value is not allowed to be a string.
For some use cases, there are good alternatives to sum(). The preferred, fast way to concatenate a sequence of strings is by calling "".join(sequence). To add floating point values with extended precision, see math.fsum(). To concatenate a series of iterables, consider using itertools.chain().

sum() 的第二個參數(shù)不允許是字符串。如果用了，會報錯：

TypeError: sum() can"t sum strings [use "".join(seq) instead]

為什么不建議使用 sum() 來拼接字符串呢？哈哈，文檔中建議使用 join() 方法，因為它更快。為了不給我們使用慢的方法，它竟特別限定不允許 sum() 的第二個參數(shù)是字符串。

文檔還建議，在某些使用場景時，不要用 sum() ，例如當以擴展精度對浮點數(shù)求和時，推薦使用 math.fsum() ；當要拼接一系列的可迭代對象時，應考慮使用 itertools.chain() 。

浮點數(shù)的計算是個難題，我曾轉(zhuǎn)載過一篇《如何在 Python 里面精確四舍五入？》，對此有精彩分析。而itertools.chain() 可以將不同類型的可迭代對象串聯(lián)成一個更大的迭代器，這在舊文《Python進階：設計模式之迭代器模式》中也有論及。

不經(jīng)意間，sum() 函數(shù)的注意事項，竟把 Python 其它的進階內(nèi)容都聯(lián)系起來了。小小的函數(shù)，竟成為學習之路上的一個樞紐。

前段時間，我還寫過 range() 、locals() 和 eval() 等內(nèi)置函數(shù)，也是通過一個問題點，而關聯(lián)出多個知識點， 獲益良多。這些內(nèi)置函數(shù)/類的魔力可真不小啊。

本文到此結(jié)束，希望對你有所幫助。

友情提示： 本公眾號內(nèi)有限時抽獎活動，送出 Python 書籍 15 本，書目有《Python數(shù)據(jù)分析與挖掘?qū)崙?zhàn)》、《Python語言程序設計》《自學Python編程基礎、科學計算及數(shù)據(jù)分析》《實用機器學習》、《Python程序員面試算法寶典》，活動結(jié)束時間為本月 20 日 18 點 ，趕快來抽獎啦！詳情請戳：抽獎送書

公眾號【Python貓】， 專注Python技術、數(shù)據(jù)科學和深度學習，力圖創(chuàng)造一個有趣又有用的學習分享平臺。本號連載優(yōu)質(zhì)的系列文章，有喵星哲學貓系列、Python進階系列、好書推薦系列、優(yōu)質(zhì)英文推薦與翻譯等等，歡迎關注哦。PS：后臺回復“愛學習”，免費獲得一份學習大禮包。