Сравнение столбца в массиве с набором значений и возврат другого набора значений из второго столбца с соответствующими индексами - PullRequest
0 голосов
/ 03 марта 2020

Итак, у меня есть массив с 2 столбцами (x, y). Мне нужно найти значения в столбце y, соответствующие некоторому другому набору чисел, скажем, [0.5, 0.5, 0.99] и вернуть значения из столбца x с такими же индексами в новую переменную.

x=np.linspace(50,70,20)
y=np.linspace(0,1,20)
c=np.zeros((2,len(x)))
x=np.around(x,3)
y=np.around(y,3)
for ii, (left, right) in enumerate(zip(x[1:], y[1:])):
    print(left, right)
    c[0, ii] = left
    c[1, ii] = right
q=[0.05,0.5,0.99]

Поэтому мне нужно сравнить c [1 ,:] с q и затем вернуть значения из c [0 ,:] с соответствующими индексами. Я пробовал for и enumerate, но не могу понять, нужно ли мне использовать итератор один или два раза (для c и q). Спасибо!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...