获取二维数组对角线上的数值
时间: 2024-01-07 15:17:20 浏览: 84
二维数组对角线.cpp
可以使用循环遍历二维数组,每次访问二维数组的行和列相同的元素,即对角线上的元素。示例代码如下:
```python
# 定义一个二维数组
matrix = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
# 遍历二维数组,获取对角线上的元素
diagonal = []
for i in range(len(matrix)):
diagonal.append(matrix[i][i])
# 打印对角线上的元素
print(diagonal) # [1, 5, 9]
```
上述代码中,使用 `range(len(matrix))` 遍历二维数组的行和列,每次访问 `matrix[i][i]` 获取对角线上的元素,并将其添加到 `diagonal` 列表中。最后打印 `diagonal` 列表即可。
阅读全文