4

numpy配列があるとします

A = [[1 2 3]
     [2 3 3]
     [1 2 3]]

そして別の配列

B = [[3 2 3]
     [1 2 3]
     [4 6 3]]

および真の値の配列:

C = [[1 4 3]
     [8 7 3]
     [4 10 3]]

ここで、配列 D を作成します。その要素は A または B のいずれかから導出されます。条件は、配列 C の各要素の最も近い値です。

これを行うためのpythonicの方法はありますか?現在、ループを使用しています

4

1 に答える 1

7
>>> K = abs(A - C) < abs(B - C)  # create array of bool
[[True, False, False],
 [True,  True, False],
 [False, False, False]]
>>> D = where(K, A, B)     # get elements of A and B respectively
于 2013-09-10T10:47:50.550 に答える