NumPy中怎么创建子数组

   2024-10-18 4150
核心提示:要在NumPy中创建子数组,可以使用切片操作符:来选择数组的子集。以下是一些示例代码:创建一个一维数组并选择子数组:import num

要在NumPy中创建子数组,可以使用切片操作符:来选择数组的子集。以下是一些示例代码:

创建一个一维数组并选择子数组:
import numpy as nparr = np.array([1, 2, 3, 4, 5])sub_arr = arr[1:4] # 选择索引为1到3的子数组print(sub_arr)
创建一个二维数组并选择子数组:
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])sub_arr = arr[:2, 1:] # 选择前两行和第二列之后的子数组print(sub_arr)
使用布尔索引选择子数组:
arr = np.array([1, 2, 3, 4, 5])mask = np.array([True, False, True, False, True])sub_arr = arr[mask] # 选择mask中为True的元素组成的子数组print(sub_arr)

通过这些示例,您可以根据需要创建和选择NumPy数组的子数组。

 
举报打赏
 
更多>同类维修大全
推荐图文
推荐维修大全
点击排行

网站首页  |  关于我们  |  联系方式网站留言    |  赣ICP备2021007278号