Pandas карта эквивалентна Numpy - PullRequest
0 голосов
/ 29 мая 2020

В настоящее время у меня есть эти два массива,

import numpy as np

arr = np.array(["Aaple", "Orange", "Banana"]).reshape(-1,1)

arr2 = np.array([["Strawberry", "Red"], ["Aaple", "Red"], ["Orange", "Orange"], ["Pear", "Yello"], ["Banana", "Yellow"]])

# they print as this
[['Aaple']
 ['Orange']
 ['Banana']]

[['Strawberry' 'Red']
 ['Aaple' 'Red']
 ['Orange' 'Orange']
 ['Pear' 'Yello']
 ['Banana' 'Yellow']]

Я хотел бы использовать эквивалент карты pandas, где я бы получил конечный результат

 [['Aaple' 'Red']
 ['Orange' 'Orange']
 ['Banana' 'Yellow']]

Где я сопоставить arr[:,1] на arr. Как бы я это сделал, я использовал np.where и довольно застрял.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...