使用Numpy Python限制矩阵每行中的元素数量

import numpy as np

# 创建一个4x4的矩阵
matrix = np.array([[1, 2, 3, 4],
                   [5, 6, 7, 8],
                   [9, 10, 11, 12],
                   [13, 14, 15, 16]])

# 限制每行只取前两个元素
limited_matrix = matrix[:, :2]

print(limited_matrix)

[[ 1  2]
 [ 5  6]
 [ 9 10]
 [13 14]]

import numpy as np

# 创建一个4x4的矩阵
matrix = np.array([[1, 2, 3, 4],
                   [5, 6, 7, 8],
                   [9, 10, 11, 12],
                   [13, 14, 15, 16]])

# 将每行限制为只有2个元素
limited_matrix = np.resize(matrix, (4, 2))

print(limited_matrix)

[[ 1  2]
 [ 3  4]
 [ 5  6]
 [ 7  8]]