get index of element in numpy array python
itemindex = numpy.where(array==item)
get index of element in numpy array python
itemindex = numpy.where(array==item)
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
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]
Copyright © 2021 Codeinu
Forgot your account's password or having trouble logging into your Account? Don't worry, we'll help you to get back your account. Enter your email address and we'll send you a recovery link to reset your password. If you are experiencing problems resetting your password contact us