2016-11-05 45 views
0

给出的播放列表:的Python - 简化代码列表理解

energy_playlist = [{u'track': u'Nude', u'feature': u'energy', u'value': 0.342, u'artist': u'Radiohead'}, {u'track': u'Kaleidoscope', u'feature': u'energy', u'value': 0.285, u'artist': u'Coldplay'}, {u'track': u'Faust Arp', u'feature': u'energy', u'value': 0.289, u'artist': u'Radiohead'}, {u'track': u'Running Up That Hill', u'feature': u'energy', u'value': 0.356, u'artist': u'Placebo'}, {u'track': u'Codex', u'feature': u'energy', u'value': 0.128, u'artist': u'Radiohead'}, {u'track': u'True Love Waits', u'feature': u'energy', u'value': 0.132, u'artist': u'Radiohead'}, {u'track': u'Asleep - 2011 Remastered Version', u'feature': u'energy', u'value': 0.255, u'artist': u'The Smiths'}, {u'track': u'Glass Eyes', u'feature': u'energy', u'value': 0.11, u'artist': u'Radiohead'}, {u'track': u'Pyramid Song', u'feature': u'energy', u'value': 0.335, u'artist': u'Radiohead'}, {u'track': u'Life On Mars? - 2015 Remastered Version', u'feature': u'energy', u'value': 0.384, u'artist': u'David Bowie'}, {u'track': u'Exit Music (For a Film)', u'feature': u'energy', u'value': 0.276, u'artist': u'Radiohead'}, {u'track': u'Videotape', u'feature': u'energy', u'value': 0.384, u'artist': u'Radiohead'}, {u'track': u'High And Dry', u'feature': u'energy', u'value': 0.383, u'artist': u'Radiohead'}] 

tempo_playlist = [{u'track': u'Codex', u'feature': u'tempo', u'value': 58.993, u'artist': u'Radiohead'}, {u'track': u'Pyramid Song', u'feature': u'tempo', u'value': 77.078, u'artist': u'Radiohead'}, {u'track': u'Videotape', u'feature': u'tempo', u'value': 77.412, u'artist': u'Radiohead'}] 

我能找到这样的匹配tracks

for d1 in energy_playlist: 
    for d2 in tempo_playlist: 
     if d1['track'] == d2['track']: 
      print (d2['track']) 

如何管理在一行中使用list comprehension做相同的操作,分配给variablefinal_playlist

+0

不能在列表理解 –

+0

我知道打印,但我可以把它分配给一个变量,并打印出来,对不对? –

+0

你可以把理解分配给一个变量,是的。你有没有试过写这个? –

回答

2

你的整个代码转换成列表解析

final_playlist = [d2['track'] for d1 in energy_playlist for d2 in tempo_playlist if d1['track'] == d2['track']] 
2

这有帮助吗?

energy_tracks = [p["track"] for p in energy_playlist] 
tempo_tracks = [p["track"] for p in tempo_playlist] 

print set(energy_tracks).intersection(tempo_tracks) 

好吧,在一行中,你现在想要它:-D?我问为什么,只是为了好玩...

result = set(p["track"] for p in energy_playlist).intersection(p["track"] for p in tempo_playlist) 

其实,这一个班轮是也许有点比三班轮更快以上,作为曲目的清单中没有明确保存在内存中,但作为set对象使用的迭代器。

+0

V我正在考虑单个列表comphreension。 –

+0

好的,对不起,这是我在下午1点多拿到的:-D。现在我不知道如何把它变成一种理解。祝你好运。 –

2

在这里你去:

[x['track'] for x in tempo_playlist if x['track'] in [y['track'] for y in energy_playlist]] 

虽然我与其他人在一条线恶补同意比多行的可读性。

+0

多数民众赞成在我正在寻找,谢谢 –

+0

现在'temp_playlist'中的每个x现在将执行'[y ['track'] for energy_playlist]'?这样会比较慢,尽管显然你已经理解了它。我相信我的速度要快得多,但我可能是错的。 :-)。 –

+0

我知道,尽管如此,我试图训练我的可读性,并且学会在一行中阅读它,在脑中创造出不同的突触.. :-D –

1

除了一)你的代码的可读性,所以看不到一条线的需要,但最重要的B)一个衬垫不能真正改善性能:)...让我证明:

import timeit 

energy_playlist = [{u'track': u'Nude', u'feature': u'energy', u'value': 0.342, u'artist': u'Radiohead'}, {u'track': u'Kaleidoscope', u'feature': u'energy', u'value': 0.285, u'artist': u'Coldplay'}, {u'track': u'Faust Arp', u'feature': u'energy', u'value': 0.289, u'artist': u'Radiohead'}, {u'track': u'Running Up That Hill', u'feature': u'energy', u'value': 0.356, u'artist': u'Placebo'}, {u'track': u'Codex', u'feature': u'energy', u'value': 0.128, u'artist': u'Radiohead'}, {u'track': u'True Love Waits', u'feature': u'energy', u'value': 0.132, u'artist': u'Radiohead'}, {u'track': u'Asleep - 2011 Remastered Version', u'feature': u'energy', u'value': 0.255, u'artist': u'The Smiths'}, {u'track': u'Glass Eyes', u'feature': u'energy', u'value': 0.11, u'artist': u'Radiohead'}, {u'track': u'Pyramid Song', u'feature': u'energy', u'value': 0.335, u'artist': u'Radiohead'}, {u'track': u'Life On Mars? - 2015 Remastered Version', u'feature': u'energy', u'value': 0.384, u'artist': u'David Bowie'}, {u'track': u'Exit Music (For a Film)', u'feature': u'energy', u'value': 0.276, u'artist': u'Radiohead'}, {u'track': u'Videotape', u'feature': u'energy', u'value': 0.384, u'artist': u'Radiohead'}, {u'track': u'High And Dry', u'feature': u'energy', u'value': 0.383, u'artist': u'Radiohead'}] 
tempo_playlist = [{u'track': u'Codex', u'feature': u'tempo', u'value': 58.993, u'artist': u'Radiohead'}, {u'track': u'Pyramid Song', u'feature': u'tempo', u'value': 77.078, u'artist': u'Radiohead'}, {u'track': u'Videotape', u'feature': u'tempo', u'value': 77.412, u'artist': u'Radiohead'}] 

def foo1(): 
    ret = [] 
    for d1 in energy_playlist: 
     for d2 in tempo_playlist: 
      if d1['track'] == d2['track']: 
       ret.append(d1["track"]) 
    return ret 

def foo2(): 
    ret = [] 
    for d1 in energy_playlist: 
     for d2 in tempo_playlist: 
      if d1['track'] == d2['track']: 
       pass 
    return None 

def foo3(): 
    return [d2['track'] for d1 in energy_playlist for d2 in tempo_playlist if d1['track'] == d2['track']] 


def bar(): 
    tempo_tracks = [i["track"] for i in tempo_playlist] 
    return [i["track"] for i in energy_playlist if i["track"] in tempo_tracks] 

print("foo1:", timeit.timeit(foo1)) 
print("foo2:", timeit.timeit(foo2)) 
print("foo3:", timeit.timeit(foo3)) 
print("bar:", timeit.timeit(bar)) 

# foo1: 5.550314342981437 
# foo2: 5.025758317991858 
# foo3: 5.3763819159939885 
# bar: 2.86007208598312