取numpy数组的某几行某几列方法

这个操作在numpy数组上的操作感觉有点麻烦,但是也没办法。

例如

a = [[1,2,3],

[4,5,6],

[7,8,9]]

取 a 的 2 3 行, 1 2 列

c=[1,2]

d =[0,1]

若写为

b = a[c,d]

output:

[4 8]

取的是 第二行第一列 和第三行第二列的数据

这并不是我们想要的结果。

正确做法是:

b = a[c]先取想要的行数据

b = b[:,d]

print(b)

output:

[[4 5]

[7 8]]

这才是我们想要的结果。必须要经过这两步才能完成。

以上这篇取numpy数组的某几行某几列方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

以上是 取numpy数组的某几行某几列方法 的全部内容, 来源链接: utcz.com/z/360060.html

回到顶部