欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  后端开发

怎样取numpy数组指定行列

程序员文章站 2022-03-22 19:44:58
...
这次给大家带来怎样取numpy数组指定行列,取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]]

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

相信看了本文案例你已经掌握了方法,更多精彩请关注其它相关文章!

推荐阅读:

完美解决python2.7无法使用pip

python怎么逐行读写txt文件

以上就是怎样取numpy数组指定行列的详细内容,更多请关注其它相关文章!