Сравнение элементов списка со столбцом в массиве numpy

У меня есть список python следующим образом.

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

и массив numpy следующим образом.

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

Я хочу сравнить каждый элемент в этом list с первым столбцом array, а затем создать new_array с совпавшими элементами. Таким образом, типичный вывод будет следующим.

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

Я попробовал следующий код.

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 для i в in_array, если i[0] в in_list]

— 21.06.2019 02:57

python numpy

21.06.2019 02:54

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

649

Перейти к ответу Данный вопрос помечен как решенный

Ответы 2

Это действительно пустой массив или просто «массив» по имени?

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')

но не будем притворяться и составим список:

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']

Понимание списка прекрасно справляется со своей задачей:

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]]

21.06.2019 03:02

Ответ принят как подходящий

Сделайте понимание этого списка:

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)

21.06.2019 03:02

Другие вопросы по теме

Как замаскировать шкалу серого изображения с помощью нарезки массива numpy

Как эффективно отображать данные между временными рядами в python

Массив Numpy как элемент управления фреймами данных pandas

Как я могу использовать lambdify для оценки моей функции?

Способ получить локальные минимумы и максимумы вектора

Np.where возвращает пустой индекс для списка кортежей

Почему реализация весов классов делает модель хуже

Есть ли шанс в Python3 создать матрицу массивов массивов?

Удалить строку с порогом или категорией и сохранить в несколько CSV в пандах

Поведение np.where() для 2d-массивов

Сравнение элементов списка со столбцом в массиве numpy

Ответы 2

Другие вопросы по теме

Похожие вопросы