Comparing list elements with a column in numpy array

Question

I have a python list as follows.

list=['M', 'R', 'W']

and a numpy array as follows.

array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4], ['Y', 115.8, -160.4]]

I want to compare each element in this list with the first column of array and then create a new_array with the elements matched. So the typical output would be as follows.

new_array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4]]

I tried the following code.

new_array=np.empty((4,3)) 

for i in range (0,len(list)):
           if list[i]==array[i; 0:1]
                new_array=np.append(new_array, (array[i,1:4].reshape(4,3)), axis=0)

new_array = [i for i in in_array if i[0] in in_list]

Osman Mamun
– Osman Mamun

2019-06-21 00:57:26 +00:00
Commented Jun 21, 2019 at 0:57 — Osman Mamun
– Osman Mamun, Commented Jun 21, 2019 at 0:57

arajshree · Accepted Answer · 2019-06-21 01:02:52Z

2

Do this list comprehension:

list=['M', 'R', 'W']
array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4], ['Y', 115.8, -160.4]]

new_array = [x for x in array if x[0] in list]
print(new_array)

answered Jun 21, 2019 at 1:02

arajshree

6244 silver badges13 bronze badges

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

Comments

hpaulj · Accepted Answer · 2019-06-21 01:02:11Z

Is that really a numpy array, or just a 'array' by name?

In [238]: np.array([['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4], ['Y', 115.8, -160.4]
     ...: ])                                                                                         
Out[238]: 
array([['M', '360.0', '360.0'],
       ['R', '135.9', '360.0'],
       ['W', '101.4', '-125.4'],
       ['Y', '115.8', '-160.4']], dtype='<U6')

but let's not pretend, and make a list:

In [239]: alist = [['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4], ['Y', 115.8, -160.4]]
     ...:                                                                                            
In [240]: alist                                                                                      
Out[240]: 
[['M', 360.0, 360.0],
 ['R', 135.9, 360.0],
 ['W', 101.4, -125.4],
 ['Y', 115.8, -160.4]]
In [241]: list1 = ['M','R','W']

A list comprehension does the job nicely:

In [243]: [item for item in alist if item[0] in list1]                                               
Out[243]: [['M', 360.0, 360.0], ['R', 135.9, 360.0], ['W', 101.4, -125.4]]

Collectives™ on Stack Overflow

Comparing list elements with a column in numpy array

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related