В настоящее время у меня есть эти два массива,
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
и довольно застрял.