質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.50%
Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

Q&A

解決済

1回答

19180閲覧

2次元配列で特定の列を検索し、行の番号と数を得たい

wwuuccccii

総合スコア13

Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

0グッド

0クリップ

投稿2017/09/10 06:37

Python3.6.2を使用しております。
2次元配列で特定の列を検索し、その値をもつ行の番号と数を得たいです。
例えば
list=[[0 2 0 0 0 0]
[5 1 0 0 0 0]
[2 1 0 0 0 0]
[7 1 0 0 0 0]]
という配列があり2列目に1がある行の
番号を[1 2 3]、数を3という形で返してほしいのです。

試したには試したのですが、
numpy.where(list[:,1]==1)
len(numpy.where(list[:,1]==1))

(array([1, 2, 3], dtype=int64),)
1

このように上手くいきません。

補足
numpy.where(list[:,1]==1)[0]

[1 2 3]

でなんとかなるのですが、これは二次元配列の中身ですよね。
一次元配列を得る方法を教えていただけないでしょうか。

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答1

0

ベストアンサー

numpy.where(list[:,1]==1)[0]

[1 2 3]

でなんとかなるのですが、これは二次元配列の中身ですよね。

いえ。1列目の値が1である行位置(の1次元配列)で合っています。
.whereの結果はタプルで返るので.where(list[:,1]==1)[0]でよいです。
よって、そのshape[0](shapeの結果もタプルなので)が条件に一致する行数になります。

Python

1import numpy as np 2a = np.array([[10,123],[20,111],[30,123],[40,222]]) 3print(a) 4r = np.where(a[:,1]==123) # 1列目の値が123の行位置を返す 5print(r) # (array([0, 2], dtype=int64),) [0,2]は行位置 6print(r[0].shape[0]) # 2(行)

投稿2017/09/10 06:54

can110

総合スコア38233

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

wwuuccccii

2017/09/10 11:35

ご回答ありがとうございます。 丁寧なご説明でとても解かり易かったです。 助かりました。
guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.50%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問