python タプル型からある要素を取り出し、新たなリストを生成する。

Question

タプル型のある順番の要素だけを取り出すコードを書きたいです。

今回、「nxのi番目とnyのj番目の数字をかけ合わせた結果aという値になった」という情報を
lというリストに格納しました。
そして、この後にaの平均値や最大値、最小値を評価をしたいので、aだけを取り出すためにmというリストを作り、
そこに格納しました。

import numpy as np
        
nx=np.array(range(2))
ny=np.array(range(3))

l=[]
m=[]
for i in range(len(nx)):
    for j in range(len(ny)):
           a=nx[i]*ny[j]
           l.append((nx[i],ny[j],a))
           m.append(a)
           

しかし、評価したいリストの数が増えると変数が増えて煩雑なコードになってしまうので
mというリストを作らず、タプル型のある要素だけ取り出して、その評価を行いたいです。

具体的には以下の結果から新たにタプルの最後の要素であるリスト　[0,1,2,1,2,3]を生成するにはどうしたらいいのでしょうか。

print(l)
 [(0, 0, 0), (0, 1, 1), (0, 2, 2), (1, 0, 1), (1, 1, 2), (1, 2, 3)]

ご回答よろしくお願いします。

Answer 1

バターンマッチもありだと思います。

l = [(0, 0, 0), (0, 1, 1), (0, 2, 2), (1, 0, 1), (1, 1, 2), (1, 2, 3)]
m = [a for (_, _, a) in l]
print(m)

Comments

[kunikuni96]

ご回答ありがとうございました。
バターンマッチというものを初めて聞き、いろいろ調べてみました。
今後は、別のコードにも生かしていきたいです。

Answer 2

lst = [e[-1] for e in l]

です。

>>> l = [(0, 0, 0), (0, 1, 1), (0, 2, 2), (1, 0, 1), (1, 1, 2), (1, 2, 3)]
>>> lst = [e[-1] for e in l]
>>> print(lst)
[0, 1, 2, 1, 2, 3]

Comments

[kunikuni96]

ご回答ありがとうございました。
いただいた解答を見て「内包表記」という方法を覚えました。
今後も活用していきたいと思います。