Q&A
インデントがめちゃくちゃですが、これで動きますか？ 間違っているなら質問文を編集して修正しておいてください。
numpy.arrayなどではなく、単純なリスト(list=[][])があります。このlistのある範囲において、ある数字が格納されているインデックスを、新たなリストに格納したいです。
2100×4600の行列を格納したlist[y][x]の32行から68行まで、2列目から8列目までの範囲の中で0.4が含まれているインデックスを、新たなリストにind=[x,y,x,y,x,y…]と格納したいです。現在はチェックのために(68-32)×(8-2)要素ですが、今後は125万要素の範囲を5箇所ほど設けて行いたいと考えています。
2100×4600の全てから0.4が含まているインデックスを見つけるコードは下記の様に書けました。
python
1ind=[] 2for y, row in enumerate(list): 3 for i in range(int((sum(len(v) for v in list))/len(list))): 4 try: 5 ind.append(row.index(0.4)) 6 ind.append(y) 7 break 8 except ValueError: 9 pass
上記のプログラムを回すと処理に10分ほどかかり、これ以上、処理に時間がかるることはできれば避けたいです。そんため上記のプログラムで所得したインデックスリストから、範囲以外の物を削除していくというのは、極力避けたいです。
宜しくお願い致します。
回答2件
0
ちょっと不格好ですが、以下のようにnumpy.whereを使ってもできます。
Python
1# テストデータ 2a = np.zeros((2100,4600)) 3a[32,2] = a[33,3] = 0.4 4lst = a.tolist() 5 6a = np.array(lst) 7ret = np.where(a[32:69,2:9] == 0.4) 8ind = [] 9for x,y in zip(ret[0] + 32, ret[1] + 2): 10 ind.extend((x,y)) 11 12print(ind) # [32, 2, 33, 3]
投稿2019/09/29 10:00
総合スコア38341
0
ベストアンサー
たかだか216要素の探索なので、愚直に二重ループで書くのが良いかと。indexとか使うとかえってややこしくなります。
python
1ind = [] 2for i in range(31, 68): # 行 3 for j in range(2, 7): # 列 4 if lst[i][j] == 0.4: 5 ind.append(j) 6 ind.append(i) 7
投稿2019/09/29 09:35
総合スコア30935
現在はチェックのために216要素ですが、今後は125万要素の範囲を5箇所ほど設けて行いたいと考えています。割愛してしまい申し訳ありませんでした。
この場合でも理にかなっているのでしょうか？
宜しくお願い致します。
探索範囲が125万*5ってこと？ それだとnumpyとかうまく使った方が効率がいいでしょう（場合によってはnumbaやcythonが要るかもしれないし、そもそも線形探索以外にやりようがないのかも考えるべきとか）。
あと浮動小数点数を直接比較するのは良いのでしょうか？
ありがとうございました。
試しにfor分で、100万要素を走査しましたが、私のコードより相当早く回りました。ありがとうございました。
あなたの回答
tips
プレビュー
