Numpy: массив массивов, выберите первое значение больше нуля

Question

Мои данные выглядят так:

all = [[-2,-1,0],[-1,-0.5,3],[-1,0.2,3],[0.2,1,3],[0.5,1,4]]

Мне нужно выбрать первый массив, значение которого в позиции [0] больше нуля, и вернуть мне в этом конкретном массиве значениеэлемент в положении 1. Я немного застрял, есть идеи?В моем случае это будет массив [0.2,1,3], и в этом массиве он должен вернуть мне 1.

Answer 1

Вы можете использовать условную маскировку

all_arr = np.array([[-2,-1,0],[-1,-0.5,3],[-1,0.2,3],[0.2,1,3],[0.5,1,4]])
boolean = all_arr[:,0]>0
# [False False False  True  True]

print (all_arr[boolean][0,1]) # 0 because you just need the very first occurence
# 1.0

Answer 2

Вы можете использовать next с выражением генератора, затем использовать индексирование списка:

res = next(x for x in all_arr if x[0] > 0)[1]  # 1

Не используйте all для имени переменной, это встроенное значение.

Если вы заинтересованы в оптимизации производительности с массивом NumPy, см. Эффективно вернуть индекс первого значения, удовлетворяющего условию в массиве .