Answers for "numpy find index of values"

3

get index of element in numpy array python

itemindex = numpy.where(array==item)
Posted by: Guest on April-28-2021
0

numpy find position in array

import numpy as np

heroes_1d = np.array(["Batman", "Spiderman", "Batman", "Superman"])
print(np.where(heroes_1d == 'Batman'))
# (array([0, 2]),)
print(heroes_1d[0], heroes_1d[2])
# Batman Batman


heroes_2d = np.array([
    ["Batman", "Spiderman", "Batman", "Superman"],
    ["Catwoman", "Batgirl", "Iron Man", "Batman"]
])
print(np.where(heroes_2d == 'Batman'))
# (array([0, 0, 1]), array([0, 2, 3]))
print(heroes_2d[0][0], heroes_2d[0][2], heroes_2d[1][3])
# Batman Batman Batman
Posted by: Guest on August-31-2021
-2

numpy find index of matching values

import numpy as np
# init arrays
a = np.array([1,2,3,2,3,4,3,4,5,6])
b = np.array([7,2,10,2,7,4,9,4,9,8])

#using enumerate, list comprehension and set
print([key for key, val in enumerate(a) if val in set(b)])
# output
# [1, 3, 5, 7]
Posted by: Guest on October-22-2020

Code answers related to "numpy find index of values"

Python Answers by Framework

Browse Popular Code Answers by Language