📌  相关文章
📜  Python程序交换矩阵中第一行和最后一行的元素(1)

📅  最后修改于: 2023-12-03 15:04:40.594000             🧑  作者: Mango

Python程序交换矩阵中第一行和最后一行的元素

在Python中,我们可以使用NumPy库来处理矩阵。下面是一个简单的Python程序,它可以将矩阵中的第一行和最后一行的元素进行交换。

首先,我们需要使用NumPy库来创建一个矩阵:

import numpy as np

# 创建一个3x3的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("原矩阵:\n", matrix)

输出:

原矩阵:
 [[1 2 3]
  [4 5 6]
  [7 8 9]]

我们可以使用np.shape函数来获取矩阵的维度。

# 获取矩阵的维度
rows, cols = np.shape(matrix)
print("矩阵的维度:", rows, ",", cols)

输出:

矩阵的维度: 3 , 3

接着,我们可以使用NumPy的切片功能来获取矩阵的第一行和最后一行。

# 获取矩阵的第一行和最后一行
first_row = matrix[0]
last_row = matrix[rows-1]
print("第一行:", first_row)
print("最后一行:", last_row)

输出:

第一行: [1 2 3]
最后一行: [7 8 9]

使用NumPy的切片功能交换第一行和最后一行的元素。

# 交换第一行和最后一行
matrix[0], matrix[rows-1] = matrix[rows-1], matrix[0]
print("新矩阵:\n", matrix)

输出:

新矩阵:
 [[7 8 9]
  [4 5 6]
  [1 2 3]]

完整代码:

import numpy as np

# 创建一个3x3的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("原矩阵:\n", matrix)

# 获取矩阵的维度
rows, cols = np.shape(matrix)
print("矩阵的维度:", rows, ",", cols)

# 获取矩阵的第一行和最后一行
first_row = matrix[0]
last_row = matrix[rows-1]
print("第一行:", first_row)
print("最后一行:", last_row)

# 交换第一行和最后一行
matrix[0], matrix[rows-1] = matrix[rows-1], matrix[0]
print("新矩阵:\n", matrix)

以上就是交换矩阵中第一行和最后一行的元素的Python程序。