Маскировать массив разных размеров, затем заменять элементы другого - PullRequest
Новая
       11

Маскировать массив разных размеров, затем заменять элементы другого

0 голосов
/ 17 апреля 2020

У меня есть два массива разного размера a и b, и я хочу получить новый массив того же размера, что и самый большой (a_and_b), и с элементами, которые имеют zeros везде, кроме когда разница между любым из двух элементов минимальна. Затем я хочу заменить его ненулевые элементы элементами другого массива того же размера, что и первый 

a = np.asarray([1, 3])
b = np.asarray([0.8, 0.95, 1.1, 1.25, 1.40, 1.55, 1.70, 1.85, 2.00, 2.15, 2.30, 2.45, 2.60, 2.75, 2.90, 3.05])
c = np.asarray([15.145, 18.191])

a_and_b = [0 0.95 0 0 0 0 0 0 0 0 0 0 0 0 0 3.05]
final = [0 15.145 0 0 0 0 0 0 0 0 0 0 0 0 0 18.191]

Как получить final?

1 Ответ

1 голос
/ 17 апреля 2020

Попробуйте это. 

import numpy as np

a = np.asarray([1, 3])
b = np.asarray([0.8, 0.95, 1.1, 1.25, 1.40, 1.55, 1.70, 1.85, 2.00, 2.15, 2.30, 2.45, 2.60, 2.75, 2.90, 3.05])
c = np.asarray([15.145, 18.191])

a_and_b, final = np.zeros_like(b), np.zeros_like(b)
min_ba = (abs(b.reshape(1,-1)-a.reshape(-1,1)).argmin(axis=1))
a_and_b[min_ba] = b[min_ba]
final[min_ba] = c
print(a_and_b)
print(final)

Out: 

[0.   0.95 0.   0.   0.   0.   0.   0.   0.   0.   0.   0.   0.   0.
 0.   3.05]
[ 0.    15.145  0.     0.     0.     0.     0.     0.     0.     0.
  0.     0.     0.     0.     0.    18.191]
...