Python 列表理解 - 比较 3 个列表

Question

我是列表理解的新手。目前我在比较两个列表时理解它，但现在我比较 3，我有点困惑。基本上我有这个清单：

teams= ['subteamA', 'subteamB'] 将这两个团队与基本上只是在 subteamA 和 subteamB 之间交替的 csv 文件进行比较。我的列表理解是这样的：

teamDimensions_semantic = [[ 1 if paramTeam == thisTeam else 0 for paramTeam in mySubteams] for thisTeam in teams]

print(teamDimensions_semantic)

然后输出类似于 [1,0,1,0,1,0] 的东西。

我现在的问题是，如果列表理解中有 3 个团队，我该如何做同样的事情？我将我的团队更改为 teams= [subteamA, subteamB, subteamC] 现在并更改了 csv 文件以反映这一点。但现在我希望我的 teamDimensions_semantic 看起来像 [1,0,2,1,0,2,1,0,2]。

我现在如何在列表理解行中写下它？我被困在 if 和 else 上......任何帮助将不胜感激！

Answer 1

不要对if 使用嵌套列表推导。使用index() 获取teams 列表中的位置。

teamDimensions_semantic = [teams.index(s) for s in mySubteams]