必会的List Comprehension （列表推导式）

强大的List Comprehension （列表推导式）是Python中必须知道的概念。然而对于初学者来说是最具挑战性的。掌握这个概念将会在两个方面帮助你:

- 应该写更短和更高效率的代码

- 代码应该执行的更快

List Comprehension （列表推导式）比for循环快35%，比map快45% 。
注：下面将List Comprehension （列表推导式）简写为LC

什么是LC

看下面的例子：
{ x^2: x 是小于10的自然数 }
{ x: x是小于20的偶数 }
{ x: x是在 ‘MATHEMATICS’中的元音字母 }

对于初学者乍一看缺失有点懵逼，来让我们仔细观察。
每个例子包含三个方面：迭代，过滤条件，执行过程
因此，也可以认为这个只是for循环的另一种表现形式。
总之，一个for循环这样执行：

for (set of values to iterate):  if (conditional filtering): 
    output_expression()

同样用LC形式构造一行:

 [ output_expression() for(set of values to iterate) if(conditional filtering) ]

另一个例子： { x: x是小于等于100的自然数, x的完全平方数 }
for循环写法：

for i in range(1,101):     #the iterator
   if int(i**0.5)==i**0.5: #conditional filtering
     print i               #output-expression

LC 写法：

[i for i in range(1,101) if int(i**0.5)==i**0.5]

找到一些感觉没？如果你理解了，那么LC是更简单且有效的强力工具，帮助你轻松完成很多工作。

记住：

• 1.List循环总是返回一个结果，无论你是否用到这个结果。

• 2.迭代和条件表达式可以嵌套多个实例。

• 3.甚至整个List循环都可以嵌套在另一个List循环中

• 4.多个变量可以同时迭代和操纵

应用1:平坦化矩阵（Flatten a Matrix）

目标：将一个矩阵输入，返回一个List，每一行放在一行的后面。

代码如下：

def eg1_for(matrix):
    flat = []    for row in matrix:        for x in row:
            flat.append(x)    return flatdef eg1_lc(matrix):
    return [x for row in matrix for x in row ]

让我们定义一个矩阵，测试下结果：

matrix = [ range(0,5), range(5,10), range(10,15) ]print "Original Matrix: " + str(matrix)print "FOR-loop result: " + str(eg1_for(matrix))print "LC result      : " + str(eg1_lc(matrix))

应用二：将一个句子中的元音字母去掉

目标：一个string作为输入，返回一个去除元音字母的string

代码如下：

def eg2_for(sentence):
    vowels = 'aeiou'
    filtered_list = []    for l in sentence:        if l not in vowels:
            filtered_list.append(l)    return ''.join(filtered_list)def eg2_lc(sentence):
    vowels = 'aeiou'
    return ''.join([ l for l in sentence if l not in vowels])

测试下：

sentence = 'My name is Aarshay Jain!'print "FOR-loop result: " + eg2_for(sentence)print "LC result      : " + eg2_lc(sentence)

应用三：字典推导式（ Dictionary Comprehension）

目标：两个一样长度的List作为输入，返回一个字典，其中一个key，一个作为value

代码如下：

def eg3_for(keys, values):
    dic = {}    for i in range(len(keys)):
        dic[keys[i]] = values[i]    return dicdef eg3_lc(keys, values):
    return { keys[i] : values[i] for i in range(len(keys)) }

测试：

country = ['India', 'Pakistan', 'Nepal', 'Bhutan', 'China', 'Bangladesh']
capital = ['New Delhi', 'Islamabad','Kathmandu', 'Thimphu', 'Beijing', 'Dhaka']print "FOR-loop result: " + str(eg3_for(country, capital))print "LC result      : " + str(eg3_lc(country, capital))

到现在,我们都集中在LC的第一印象,即它的简洁和可读性。但这还不止!LC在各种场景中相比表现的更快。让我们进一步探讨，这里不做详细介绍。

在数据分析中的应用

对于数据分析，LC也是非常有用。
一个快速的回顾，记住LC的结构：

 [ output_expression() for(set of values to iterate) if(conditional filtering) ]

应用4：读双重LIst

可能遇到过这样的场景，在数据集中每个列都包含一个LIst。下面我们举例说明，数据集包含两列：

• personID: 独一无二

• skills: 每个人会的运动

Lets load the dataset:import pandas as pd
data = pd.read_csv("skills.csv")print data

在这里,每个人会多种不同的运动。将这些数据永在预测模型,一个好主意通常是来创建一个新的列每个运动和将其标记为1或0。
第一步是将文本转换成可以访问列表,这样单独的条目。

#Split text with the separator ';'data['skills_list'] = data['skills'].apply(lambda x: x.split(';'))print data['skills_list']

接下来，我们需要一个独特的运动列表来确定所需的不同的列数。这可以通过理解，独特元素构成的set（集合）。

#Initialize the setskills_unq = set()#Update each entry into set. Since it takes only unique value, duplicates will be ignored automatically.skills_unq.update( (sport for l in data['skills_list'] for sport in l) )print skills_unq

注意，这里我们使用生成器表达式，这样每个值不必动态更新和存储。现在我们使用LC将生成一个与0 - 1矩阵包含5列标记相应的运动。

#Convert set to list:skills_unq = list(skills_unq)
sport_matrix = [ [1 if skill in row else 0 for skill in skills_unq] for row in data['skills_list']  ]print sport_matrix

最后一步是生用Pandas DataFrame 合成：

data = pd.concat([data, pd.DataFrame(sport_matrix,columns=skills_unq)],axis=1)print data

应用5:为多项式回归创建一列权值（Creating powers of a columns for Polynomial regression）

多项式回归算法需要多种权值相同的变量,它可以使用LC创建。多达15 - 20的相同的变量可以用在岭回归分析建模以及减少过度拟合。
让我们简单的创建一列数据集：

data2 = pd.DataFrame([1,2,3,4,5], columns=['number'])print data2

我们定义一个变量 “deg”，包含要求的程度。我们第一步是创建一个矩阵包含不同权值的‘number’ 变量。

#Define the degree:deg=6#Create the matrix:power_matrix = [ [i**p for p in range(2,deg+1) ] for i in data2['number'] ]  
print power_matrix

与前面的示例相似，现在我们将添加到dataframe。注意,在这种情况下,我们需要一个列名列表和LC可以用来容易获取它:

cols = ['power_%d'%i for i in range(2,deg+1)]
data2 = pd.concat([data2, pd.DataFrame(power_matrix,columns=cols)],axis=1)print data2

应用6:过滤列名（Filtering column names）

个人而言,我面临过很多次这种问题，为了一个预测模型预测，选择dataframe中列的一个子集，。让我们考虑这种情况的总列:

cols = ['a', 'b', 'c', 'd', 
        'a_transform', 'b_transform', 'c_transform', 
        'd_power2', 'd_power3', 'd_power4', 'd_power5',        'temp1', 'temp2']

可以理解为：

• 1.a,b,c,d：原始数据列

• 2.a_transform,b_transform,c_transform：转换后的特征，比如log，平方根等等

• 3.d_power2,d_power3,d_power4,d_power5:：相同变量的不同的权值对于多项式或者岭回归

• 4.temp1, temp2：中间变量创建执行特定的计算
  根据正在执行的分析或使用的模型，可能有不同的用处:

• 1.选择转换的变量

• 2.选择不同权值相同的变量

• 3.选择超过2个的组合

• 4.选择所有，除了临时变量
  代码如下：

col_set1 = [x for x in cols if x.endswith('transform')]
col_set2 = [x for x in cols if 'power' in x]
col_set3 = [x for x in cols if (x.endswith('transform')) | ('power' in x)]
col_set4 = [x for x in cols if x not in ['temp1','temp2']]print 'Set1: ', col_set1print 'Set2: ', col_set2print 'Set3: ', col_set3print 'Set4: ', col_set4

End Notes

如果你已经达到这一点,我相信,现在你能够欣赏LC的重要性。你应该试着把这种技术在日常实践。尽管它可能会耗费不少时间开始,但是相信我当你进步时你会十分享受。

这篇文章能对你有帮助吗？我遗漏了什么了吗？你有一些更有趣的应用吗，您认为LC会有用吗?请分享你的评论，我们一起讨论。

英语原文链接：
https://www.analyticsvidhya.com/blog/2016/01/python-tutorial-list-comprehension-examples/

作者：CodingFish
原文链接：https://www.jianshu.com/p/c52968ab9c3d