python列表替换元素_[Python]替换列表中的元素 以及列表生成式的用法

偶然看到一个这个文章，替换元素大家应该都有方法去做，但是作者用的方法之前没有接触过，感觉比较好玩，mark记录学习下。

列表元素替换

把列表中的元素直接更改、替换。

例子：表面列表aaa中的元素‘黑色’替换成‘黄色’。

aaa=['black','red','white','black']

第一种方法：

aaa=['black','red','white','black']

aaa=str(aaa)

bbb=aaa.replace("black","yellow")

print bbb

'''

['yellow', 'red', 'white', 'yellow']

'''

第二种方法：

aaa=['black','red','white','black']

bbb=['yellow' if i =='black' else i for i in aaa]

print bbb

'''

['yellow', 'red', 'white', 'yellow']

'''

第三种方法：（替换批量的元素）

aaa=['black','red','white','black']

ccc=['black','red']

bbb=['yellow' if i in ccc else i for i in aaa]

print bbb

'''

['yellow', 'yellow', 'white', 'yellow']

'''

第四种方法：（替换多个元素）

aaa=['black','red','white','black']

ccc={'black':'yellow','red':'white'}

bbb=[ccc[i] if i in ccc else i for i in aaa]

print bbb

'''

['yellow', 'white', 'white', 'yellow']

'''

第一种方法不言而喻，可以做到但是还要再处理一次因为属性是str，后面的条件写法第一次见到，查询了下才知道这种写法叫做列表生成式，很新奇哎，感觉还挺实用的😄..

列表生成式：

首先，列表生成式有两种形式：

1. [x for x in data if condition]

此处if主要起条件判断作用，data数据中只有满足if条件的才会被留下，最终生成一个数据列表。

list1 = [i for i in range(1,10) if i%2==1]

print list1

'''

[1, 3, 5, 7, 9]

'''

2. [exp1 if condition else exp2 for x in data]

此处if…else主要起赋值作用。当data中的数据满足if条件时，将其做exp1处理，否则按照exp2处理，最终生成一个数据列表。

list2=['R','G','B','W']

list3=['R','G']

list4 = [i if i in list3 else i for i in list2]

print list4

'''

['R', 'G', 'B', 'W']

'''

实例拓展：

可以用列表生成式来对列表元素过滤。

一 .大小写转换

def toUpper(L):

return [x.upper() for x in L if isinstance(x,str)]

def toLower(L):

return [x.lower() for x in L if isinstance(x,str)]

print toUpper(['heelow','worrrd',111])

print toLower(['heelow','worrrd',111])

'''

['HEELOW', 'WORRRD']

['heelow', 'worrrd']

'''

二.list 去除指定重复元素

import copy

def filterDup(L,target):

temp=copy.copy(L)

temp_L = [L.remove(target) if (L.count(target) >=2 and x == target) else x for x in temp]

for x in temp_L:

if not x:

temp_L.remove(x)

return temp_L

list5=['R','G','B','W','B','G','R','R','W','C']

list6 = filterDup(list5,'R')

print list6

'''

['G', 'B', 'W', 'B', 'G', 'R', 'W', 'C']

'''