溫馨提示×

numpy怎么獲取指定行列

小億
258
2024-01-31 14:38:28
欄目: 編程語言

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

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

import numpy as np

# 創(chuàng)建一個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