Python create new arrays after matching some values from two dimensional arrays

Question

How to search array at some selected values?. For this case I am searching the respective "direction" and "time" for the rain values at "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']

The expected arrays are like this:

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

Anyone got any idea?. What shall I do after reshaping the data like this:

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

RomanPerekhrest · Accepted Answer · 2018-05-15 05:57:38Z

2

Simply with zip() function:

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)

The output:

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

answered May 15, 2018 at 5:57

RomanPerekhrest

93.1k4 gold badges75 silver badges112 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Azam Over a year ago

@RomanPerekhrest..awesomw...happy for that..million thanks. What is the meaning of t[-1] ???...

RomanPerekhrest Over a year ago

@Azam, welcome. [-1] points to the last item of the sequence.

Collectives™ on Stack Overflow

Python create new arrays after matching some values from two dimensional arrays

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related