神奇索引

先定义如上数据结构。

Q1:取其中的某行

data[[1,2,3,4,5]]

[[ 5  6  7  8  9] 
[10 11 12 13 14]
[15 16 17 18 19]
[20 21 22 23 24]
[25 26 27 28 29]]

一开始,我不是很理解为何要选择数据的时候要用到两个[],为何data[1,2,3,4] 这种方式不可以呢?

直到我遇到这个:

data[[1,2,3,4],[1,2,3,4]] –>[ 6 12 18 24]

在numpy中,我们更多接触到的是多维数据,data[n1,n2,n3….] 指的是对第n行的选择, 对于第一个[1,2,3,4]而言,我要的是第1,2,3,4行的数据,对于第二个[1,2,3,4]而言,我要的就是1,2,3,4列了。

美哉美哉。

发表评论

电子邮件地址不会被公开。 必填项已用*标注