python矩阵列的实现示例

import numpy as np

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

# 在第二列中获得所有元素。
column_2 = matrix[:, 1]
print(column_2)  # 输出: [2 5 8]

# 修改第三列的所有元素10。
matrix[:, 2] = 10
print(matrix)
# 输出：
# [[ 1  2 10]
#  [ 4  5 10]
#  [ 7  8 10]]

# 位置1插入全为0的列。
matrix = np.insert(matrix, 1, 0, axis=1)
print(matrix)
# 输出：
# [[ 1  0  2 10]
#  [ 4  0  5 10]
#  [ 7  0  8 10]]

# 删除已经插入的第二列
matrix = np.delete(matrix, 1, axis=1)
print(matrix)
# 输出：
# [[ 1  2 10]
#  [ 4  5 10]
#  [ 7  8 10]]

# 计算每一列的和
column_sums = matrix.sum(axis=0)
print(column_sums)  # 输出：[12 15 30]

# 计算每个列的平均值
column_means = matrix.mean(axis=0)
print(column_means)  # 输出：[ 4.  5. 10.]

# 为排序打造一个新的矩阵。
matrix_to_sort = np.array([[3, 1, 7],
                           [2, 6, 4],
                           [5, 8, 9]])

# 对每个列进行排序
np.sort(matrix_to_sort, axis=0)
print(matrix_to_sort)
# 输出：
# [[2 1 4]
#  [3 6 7]
#  [5 8 9]]

# 创建一维数组作为矩阵的列
col = np.array([1, 2, 3])

# 使用广播将col添加到matrix的每个列上。
new_matrix = matrix + col[:, np.newaxis]
print(new_matrix)
# 输出：
# [[ 2  3 11]
#  [ 6  7 12]
#  [10 11 13]]