numpy

numpy怎么获取指定行列

小亿
346
2024-01-31 14:38:28
栏目: 编程语言

要在numpy中获取指定的行和列,可以使用切片操作。

以下是获取指定行列的示例:

import numpy as np

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

# 获取第一行
row_1 = matrix[0]
print(row_1)  # 输出: [1 2 3]

# 获取第一列
column_1 = matrix[:, 0]
print(column_1)  # 输出: [1 4 7]

# 获取第二行和第三行
row_2_3 = matrix[1:3]
print(row_2_3)  # 输出: [[4 5 6]
                #        [7 8 9]]

# 获取第二列和第三列
column_2_3 = matrix[:, 1:3]
print(column_2_3)  # 输出: [[2 3]
                   #        [5 6]
                   #        [8 9]]

在numpy中,切片操作可以使用冒号:来表示范围,使用逗号,来分隔行和列的范围。

0
看了该问题的人还看了