matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
result = [matrix[i][i] for i in range(len(matrix))]
print(result)