Как получить индекс элемента numpy в python?

Question

[Ilnur]

Есть код, как найти индекс по значению функцией index() я в курсе, а вот если я делаю index(max_value), то кидает ошибку

n1, n2, n3 = 3, 5, 7

	T = [[[randint(-1000, 1000) for z in range(n3)] for j in range(n2)] for i in range(n1)]
	print("Массив:", T, "\n")

	max_value = np.amax(T)
	print("Наибольший элемент массива:", max_value, "\nИндекс элемента:")

Comments

[Алексей Уколов]

кидает ошибку

Какую?

[Ilnur]

Алексей Уколов,

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

Answer 1

[dmshar]

Раз уж вы задействовали numpy, то вот так:

t1=np.random.randint(-1000, 1000,size=n1*n2*n3).reshape(n1,n2,n3)
x,y,z = np.unravel_index(t1.argmax(), t1.shape)
t1[x,y,z]

Comments

[dmshar]

sabone, А что по-вашему указывается в квадратных скобках в выражении t1[x,y,z]?? Именно то, что принято называть "индексами" по каждой из осей вашего массива.

[Ilnur]

dmshar, благодарю за ваш ответ, все получилось