Numpy中数组元素的获取

目录


先来看一下一维数组元素和二维数组元素获取的例子,代码如下:

import numpy as np 
arr1=np.array([3,10,8,7,34,11,28,72])
arr2=np.array([[8.5,6,4.1,2,0.7],[1.5,3,5.4,7.3,9],[3.2,3,3.8,3,3],[11.2,13.4,15.6,17.8,19]])
print(arr1)
print(arr2)
#一维数组元素的获取
print(arr1[[2,3,5,7]])
#二维数组元素的获取
#第2行第3列元素
print(arr2[1,2])
print(arr2[1][2])
#第3行所有元素
print(arr2[2,:])
#第2列所有元素
print(arr2[:,1])
#第2至4行,第2至5列
print(arr2[1:4,1:5])

[ 3 10  8  7 34 11 28 72]
[[ 8.5  6.   4.1  2.   0.7]
 [ 1.5  3.   5.4  7.3  9. ]
 [ 3.2  3.   3.8  3.   3. ]
 [11.2 13.4 15.6 17.8 19. ]]
[ 8  7 11 72]
5.4
5.4
[3.2 3.  3.8 3.  3. ]
[ 6.   3.   3.  13.4]
[[ 3.   5.4  7.3  9. ]
 [ 3.   3.8  3.   3. ]
 [13.4 15.6 17.8 19. ]]

通过位置索引获取一维和二维数组中的元素,在一维数组中,列表的所有索引方法,都可以使用在数组上,而且还可以将任意位置的索引组装成列表,用作对应元素的获取;在二维数组中,位置索引必须写成[row,column]的形式。如果想要获取所有的行或列元素,那么对应的行索引或列索引用英文状态的冒号表示;

如果要从数组中获取不连续的某几行或某几列,可以采用ix_函数。

import numpy as np 
arr2=np.array([[8.5,6,4.1,2,0.7],[1.5,3,5.4,7.3,9],[3.2,3,3.8,3,3],[11.2,13.4,15.6,17.8,19]])
#第1行,最后1行和第2列,第4列构成的数组
print(arr2[np.ix_([0,-1],[1,3])])
#第1行,最后1行和第2列、第3列、第4列构成的数组
print(arr2[np.ix_([0,-1],[1,2,3])])

[[ 6.   2. ]
 [13.4 17.8]]
[[ 6.   4.1  2. ]
 [13.4 15.6 17.8]]

  • 5
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
numpy数组元素周围的操作可以通过以下几种方式实现: 1. 切片操作:可以使用numpy数组的切片操作来获取数组元素的周围元素。 例如,对于一个二维数组arr,要获取第i行第j列元素周围的元素,可以使用如下切片操作: ```python arr[i-1:i+2, j-1:j+2] ``` 这将返回一个3x3的子数组,其心元素为arr[i,j],周围的8个元素为该子数组的其余元素。 2. 使用numpy.pad()函数:numpy.pad()函数可以用来在数组的边缘添加一个或多个值,从而扩展数组的大小。可以使用该函数来添加额外的行和列,然后通过索引访问周围的元素。 例如,对于一个二维数组arr,要获取第i行第j列元素周围的元素,可以使用如下代码: ```python padded_arr = np.pad(arr, ((1, 1), (1, 1)), mode='constant') surrounding = padded_arr[i:i+3, j:j+3] ``` 这将在数组的边缘添加一行和一列,并使用常量值填充这些额外的元素。然后可以使用切片操作来获取心元素周围的元素。 3. 使用numpy.roll()函数:numpy.roll()函数可以用来沿着给定轴滚动数组的元素。可以使用该函数来将数组的行和列进行滚动,从而获取周围的元素。 例如,对于一个二维数组arr,要获取第i行第j列元素周围的元素,可以使用如下代码: ```python rows, cols = arr.shape row_indices = np.arange(i-1, i+2) % rows col_indices = np.arange(j-1, j+2) % cols surrounding = arr[row_indices][:, col_indices] ``` 这将将第i行向上和向下滚动一行,并将第j列向左和向右滚动一列,从而获取心元素周围的元素。使用模运算可以确保在数组的边缘滚动时正确处理索引。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值