溫馨提示×

NumPy怎么從現(xiàn)有數(shù)組中提取子集

小億
114
2024-05-13 11:30:14
欄目: 編程語言

要從現(xiàn)有數(shù)組中提取子集，可以使用NumPy中的切片操作。以下是一個示例：

import numpy as np

# 創(chuàng)建一個示例數(shù)組
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 從數(shù)組中提取子集
subset = arr[:2, 1:]

print(subset)

在上面的示例中，我們創(chuàng)建了一個3x3的數(shù)組arr，并使用切片操作[:2, 1:]提取了子集。這里[:2]表示提取前兩行，[1:]表示提取從第二列開始的所有列。最終得到的子集是一個2x2的數(shù)組：

[[2 3]
 [5 6]]

通過調(diào)整切片操作的參數(shù)，可以提取想要的子集。

0 踩