Use the map to locate and find a safe escape route in escape road 2.

Buongiorno! Sì, puoi utilizzare il comando np.fill_diagonal insieme a un'operazione di slicing per riempire la prima superdiagonale di una matrice NumPy. Ecco come puoi fare:


python


import numpy as np

# Definizione della matrice
A = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])

# Valori da inserire nella prima superdiagonale
new_values = np.array([10, 11]) # Sostituisci [2, 6] con [10, 11]

# Slicing per la prima superdiagonale e assegnazione dei nuovi valori
A[np.diag_indices_from(A, k=1)] = new_values

print(A)

Spero che questo ti sia utile!