Python создает новые массивы после сопоставления некоторых значений из двухмерных массивов

Question

Как искать массив по некоторым выбранным значениям?Для этого случая я ищу соответствующие «направление» и «время» для значений дождя в «5».

direction=['200','250','180','200','300','270','005','080']
time=['0000','0030','0100','0130','0200','0230','0300','0300']
rain=['15','20','100','5','23','12','5','30']

Ожидаемые массивы такие:

new array=[['200','0130','5'],['005','0300','5']]

У кого-нибудь есть идеи ?.Что мне делать после изменения данных следующим образом:

data3=np.zeros((len(direction1),3),dtype='object')
data3[:,0]=direction1
data3[:,1]=time1
data3[:,2]=rain

Answer 1

Просто с функцией zip():

direction = ['200','250','180','200','300','270','005','080']
time = ['0000','0030','0100','0130','0200','0230','0300','0300']
rain = ['15','20','100','5','23','12','5','30']

result = [list(t) for t in zip(direction, time, rain) if t[-1] == '5']
print(result)

Выход:

[['200', '0130', '5'], ['005', '0300', '5']]