ほとんどの要素値が0であるリストで、要素値が0でない部分の場所の探し方

Question

実現したいこと

ほとんどの要素値が0のリストがあります。要素値が0ではない行番号、列番号を知りたいです。二次元の場合は行の大きさ、列の大きさが分かれば「該当のソースコード」で一応実現できました。

二次元より大きい場合を見据えて、もっと良いコードの書き方があるかもしれないと思ったので、もしありましたら教えてください。

sample.csvは以下の通りです。
1 0 0
2 0 0
0 0 3

該当のソースコード

import pandas as pd
import csv

lst = pd.read_csv("sample.csv",header=None).values.tolist()
print(lst)#[[1, 0, 0], [2, 0, 0], [0, 0, 3]]

for j in range(0,3,1):
    for i in range(0,3,1):
        if lst[j][i] != 0:
            print('j='+str(j)) 
            print('i='+str(i)) 
            print(str(lst[j][i])+'\n') 

試したこと

「リストの要素値が0である行番号、列番号を知りたい」で検索しても、良いページが出てきませんでした

Comments

[poto568]

想定しているn次元の配列をcsvで表現してください。

[yyicp]

コメントありがとうございます。

14×27×27の3次元リストで、この中に要素値が0出ない部分が1～3個あります。（個数はリストの中を見るまで分かりません）

melian様の方法で解決しました。

Answer 1

numpy.nonzero() を利用するとよいかと思います。

import numpy as np

nz = [i.tolist() for i in np.nonzero(lst)]
for j, i in zip(*nz):
    print(f'{j=}') 
    print(f'{i=}') 
    print(lst[j][i], end='\n\n')

# j=0
# i=0
# 1
# 
# j=1
# i=0
# 2
# 
# j=2
# i=2
# 3

Comments

[yyicp]

できました。
どうもありがとうございました。