numpyで二つの二次元配列よりそれぞれ1列目の値を探索し、列の値が同じ場合はその行を抽出し、配列を再構成したい。

Question

numpyで二つの二次元配列よりそれぞれ1列目の値を探索し、列の値が同じ場合はその行を抽出し、配列を再構成したいです。

二つの二次元配列の1列目を参照し、同じ値の要素を見つけ出します。
見つけ出したその要素と同じ行の行列を抽出(例えば2行1列目の要素の値が同じであれば、2行目すべてを抽出)したいです。

例えば
a =[[1 10 11 12]
[2 13 14 15]
[3 16 17 18]
[4 19 20 21]]

b= [[2 20 21 22]
[3 23 24 25]
[4 26 27 28]
[5 29 30 31]
[6 32 33 34]]

この二つの配列の1列目を参照して(2,3,4が同じ値のため、その行全体を抽出)

a =[[2 13 14 15]
[3 16 17 18]
[4 19 20 21]]

b= [[2 20 21 22]
[3 23 24 25]
[4 26 27 28]]

のような処理を行いたいです。

参照するのは1列目のみで、2~4列目はただコピーするだけにしたいです。

発生している問題・エラーメッセージ

処理自体は回るのですが、以下のエラーメッセージが出て、すべて0を返してしまいます。

__main__:1: DeprecationWarning: elementwise comparison failed; this will raise an error in the future.

該当のソースコード

c = np.where(a[:,0] == b[:,0], a, False)
                         

試したこと

common_element = np.intersect1d(a,b)
c = np.where(common_element == b[:,0], a, False)

このようなコードで、a,bの共通項を抽出することはできたのですが、全く同じエラーを返してしまいます。

Comments

[meg_]

> 二つの二次元配列から同じ要素とその行を抽出
条件がよく分かりません。もう少し説明してもらえませんか？

Answer 1

おそらく質問者さんがやりたかったのはこういうことでしょう。

import numpy as np

a = np.array([[1, 10, 11, 12],
              [2, 13, 14, 15],
              [3, 16, 17, 18],
              [4, 19, 20, 21]])
b = np.array([[2, 20, 21, 22],
              [3, 23, 24, 25],
              [4, 26, 27, 28],
              [5, 29, 30, 31],
              [6, 32, 33, 34]])
common_element = np.intersect1d(a[:, 0], b[:, 0])

a[np.isin(a[:, 0], common_element)]
# array([[ 2, 13, 14, 15],
#        [ 3, 16, 17, 18],
#        [ 4, 19, 20, 21]])

b[np.isin(b[:, 0], common_element)]
# array([[ 2, 20, 21, 22],
#        [ 3, 23, 24, 25],
#        [ 4, 26, 27, 28]])

[1 2 3 4]から[2 3 4]と重なる値を抜き出すにはnp.isin()（またはnp.in1d()）を使います。
numpy.isin — NumPy Manual
配列 == 配列は配列の要素ごとの比較を行おうとしますが、[1 2 3 4]と[2 3 4]は長さが異なり要素ごとの比較ができないため、警告が出ます。

Comments

[Kevin1028]

ありがとうございます！
このような関数があったのですね・・・
もっと勉強します！

Answer 2

各配列の1列目の要素から重複するものを抽出し、それらが各配列の何行目にあたるかで行を抽出しました。

import numpy as np

a = np.array([[1,10,11,12],
[2,13,14,15],
[3,16,17,18],
[4,19,20,21]])

b = np.array([[2,20,21,22],
[3,23,24,25],
[4,26,27,28],
[5,29,30,31],
[6,32,33,34]])

c = [i for i in a[:,0] if i in b[:,0]]
d = a[[i for i in range(len(a[:,0])) if a[:,0][i] in  c],:]
e = b[[i for i in range(len(b[:,0])) if b[:,0][i] in  c],:]

print(d)
#[[ 2 13 14 15]
# [ 3 16 17 18]
# [ 4 19 20 21]]
print(e)
#[[ 2 20 21 22]
# [ 3 23 24 25]
# [ 4 26 27 28]]

Comments

[Kevin1028]

できました！
大変助かりました！本当にありがとうございます！