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

yipeiwu_com7年前Python基础

这个操作在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数组的某几行某几列方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

对pytorch网络层结构的数组化详解

最近再写openpose,它的网络结构是多阶段的网络,所以写网络的时候很想用列表的方式,但是直接使用列表不能将网络中相应的部分放入到cuda中去。 其实这个问题很简单的,使用module...

谈谈Python中的while循环语句

谈谈Python中的while循环语句

前言 python中有两种循环,while和for,两种循环的区别是,while循环之前,先判断一次,如果满足条件的话,再循环,for循环的时候必须有一个可迭代的对象,才能循环,比如说得...

python实现倒计时小工具

本文实例为大家分享了python实现倒计时小工具的具体代码,供大家参考,具体内容如下 #!/usr/bin/env python # coding=utf-8 import th...

Django objects的查询结果转化为json的三种方式的方法

Django objects的查询结果转化为json的三种方式的方法

第一种方式: 利用seriallizers 这个方法,官网的解释说:将复杂的数据结构变成json、xml或者其他的格式 import json from django.core...

跟老齐学Python之画圈还不简单吗?

跟老齐学Python之画圈还不简单吗?

在python中,循环有一个语句:for语句。 简单的for循环例子 >>> hello = "world" >>> for i in hello...