当前位置:首页 >> 脚本专栏

python [:3] 实现提取数组中的数

搜索答案搜索不到,自己试了一把.

首先生成一维数组

a =np.array([1,2,3,4,5,6,7,8,9])
> print a
[1 2 3 4 5 6 7 8 9]

取数组前3个值

b =a[:3]
> print b
[1 2 3]

取前3个以后的值

b =a[3:]
> print b
[4 5 6 7 8 9]

取数组的后3个值

b =a[-3:]
> print b
[7 8 9]

取数组后3个以前的值

b =a[:-3]
> print b
[1 2 3 4 5 6]

所以-号表示方向,从前取还是从后取,与数字配合使用,:表示所有的意思.

对于二维的数组有同样的效果,只是取的方法要考虑的行或列

a=np.eye(5)
> b=a[:,:3]
> print a
[[1. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0.]
 [0. 0. 1. 0. 0.]
 [0. 0. 0. 1. 0.]
 [0. 0. 0. 0. 1.]]
> print b
[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]
 [0. 0. 0.]
 [0. 0. 0.]]

又取行又取列

b=a[:3,:3]
> print b
[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]
 

以上这篇python [:3] 实现提取数组中的数就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。