python3 2次元リスト 複数 インデックス

Question

li=[[1,1,1,1],[2,2, 5,2],[100, 1, 0, 2,]]
ans=[]
for line_1,lis in enumerate(li):
    for line_2,ind in enumerate(lis):
        if ind ==2:
            ans.append([line_1,line_2])
print(ans) #[[1, 0], [1, 1], [1, 3], [2, 3]]

2次元リストから、2のインデックスを取得しています。
下記を内包表記とするには、どうしたらいいでしょうか？

    for line_2,ind in enumerate(lis):
        if ind ==2:
            ans.append([line_1,line_2])

Answer 1

li=[[1, 1, 1, 1],
    [2, 2, 5, 2],
    [100, 1, 0, 2]]

ans = [[i,j]
       for i,line in enumerate(li)
       for j, n in enumerate(line)
       if n == 2]

print(ans)  # => [[1, 0], [1, 1], [1, 3], [2, 3]]

基本的な考え方は、最初の式以外のforとifは普通のループと同じ順番で並んでいるということです。

とはいえ、ドキュメントや解説ページを見たり、自分で色々書き換えて動かしてみないと何をやっているのかよくわからないと思います。この書き方でどんどん深くしていくのは避けるべき作法とされています。実際、この形でforが2つのコードはたまに見かけますが、3つ以上は（ネタとして書かれたコード以外では）見たことがありません。

参考：
5. データ構造 — Python 3.6.5 ドキュメント - 5.1.3. リストの内包表記
Pythonでリストの内包表記を使って2次元リストを1次元リストにする - こんにちはこんにちはmonmonです！

Comments

[zeitaku_fire]

ありがとうございます。
参考のURL拝見しました。
疑問に思っていた点は、ドキュメントで解消できました。お手数おかけいたしました。

Answer 2

下記を内包表記とするには、どうしたらいいでしょうか？

ans = [
    [line_1, line_2]
    for line_1, row in enumerate(li)
    for line_2, elm in enumerate(row)
    if elm == 2
]

ただ、さすがにこれは読みづらいので、適当に処理を分けた方が良いです。

---

こんな感じに組んでみても面白いかもしれませんね。

lst = [[1, 1, 1, 1], [2, 2, 5, 2], [100, [1, 2, 4], 0, 2], 2]
print(lst)

def _recursive_enumerate_inner(src, indice):
    if isinstance(src, (list, tuple)):
        for i, inner in enumerate(src):
            yield from _recursive_enumerate_inner(inner, indice + [i])
    else:
        yield indice, src

def recursive_enumerate(src):
    yield from _recursive_enumerate_inner(src, [])

def recursive_index(src, purpose):
    return (
        indice
        for indice, elem in recursive_enumerate(src)
        if elem == purpose
    )

purpose = 2
for ans in recursive_index(lst, purpose):
    print(f'{purpose} at {ans}.')

実行結果 Wandbox

[[1, 1, 1, 1], [2, 2, 5, 2], [100, [1, 2, 4], 0, 2], 2]
2 at [1, 0].
2 at [1, 1].
2 at [1, 3].
2 at [2, 1, 1].
2 at [2, 3].
2 at [3].

Comments

[zeitaku_fire]

ありがとうございます。
疑問解決しました。
defの使い方は、まだまだなので、頂いたコード参考にさせて頂きます。

Answer 3

Numpy使って遊んでみました。

import numpy as np

li = np.array([[1,1,1,1],[2,2, 5,2],[100, 1, 0, 2,]])


x = np.where(li==2)[0]
y = np.where(li==2)[1]

ans = [[a, b] for a, b in zip(x, y)]
print(ans)
"""
[[1, 0], [1, 1], [1, 3], [2, 3]]
"""

Comments

[zeitaku_fire]

numpyでの解答ありがとうございます。
2次元リストを扱うなら、numpyが高速で良い。
というのを見たことあるのですが、実際には全然使ったことがなくて、頂いたコード参考にさせて頂きます。