Найти диагональ без цикла или диагонали

Question

Я пытался найти диагональ матрицы b без использования цикла или диагонали.Я получил ошибку: 'numpy.ndarray' object has no attribute 'index'.Не уверен, как это исправить.

b = np.random.randint(low = 1, high = 11, size = (10,10))

print(list(map(lambda x: x[a.index(x)], a)))

Answer 1

Объект numpy ndarray не ведет себя точно так же, как список python, в частности, как указывает ошибка, у него нет функции index.Вот один из способов обойти это:

Сначала вы можете преобразовать b из numpy ndarray в стандартный список python следующим образом:

b = b.tolist()

Затем код, который вы написалибудет работать.

print(list(map(lambda x: x[b.index(x)], b)))