繁体   English   中英

提高列表理解力

[英]Improving a List Comprehension

下面的列表理解实现了预期的结果/输出:

  1. 列表中的每个元组按顺序索引
  2. 这些名字按西装组合在一起
  3. 名称顺序一致

代码(纸牌玩具示例):

Suits = ["Hearts"]*3 +["Diamonds"]*3
Names = ["Two","Ten","King"]

z = [(i,Names[(i-1)%3],Suits[((i-1)%6)]) for i in range(1,7)]

Output z:

[(1, 'Two', 'Hearts'),
 (2, 'Ten', 'Hearts'),
 (3, 'King', 'Hearts'),
 (4, 'Two', 'Diamonds'),
 (5, 'Ten', 'Diamonds'),
 (6, 'King', 'Diamonds')]

它起作用是因为原始的Suits列表从["Hearts","Diamonds"] to["Hearts"]*3 +["Diamonds"]*3.

这感觉做作; 虽然它有效,但我正在寻找更好的解决方案。

使用enumerateitertools.product

from itertools import product

ranks = ["Two", "Ten", "King"]
suits = ["Hearts", "Diamonds"]
z = [(i, r, s) for i, (s, r) in enumerate(product(suits, ranks), start=1)]

Output:

[(1, 'Two', 'Hearts'),
 (2, 'Ten', 'Hearts'),
 (3, 'King', 'Hearts'),
 (4, 'Two', 'Diamonds'),
 (5, 'Ten', 'Diamonds'),
 (6, 'King', 'Diamonds')]

您可以使用itertools.productenumerate

from itertools import product

suits = ["Hearts", "Diamonds"]
sames = ["Two","Ten","King"]

cards = [(index, suit, rank) for index, (suit, rank) in enumerate(product(sames, suits), start=1)]

print(cards)

您可以使用带有枚举的两级嵌套理解来获得编号:

Suits = ["Hearts","Diamonds"]
Names = ["Two","Ten","King"]


r = [(j+i,n,s) for j,s in enumerate(Suits) 
               for i,n in enumerate(Names,j*len(Suits)+1) ]

print(*r,sep="\n")
(1, 'Two', 'Hearts')
(2, 'Ten', 'Hearts')
(3, 'King', 'Hearts')
(4, 'Two', 'Diamonds')
(5, 'Ten', 'Diamonds')
(6, 'King', 'Diamonds')

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM