取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设计】。

相关文章

python中dict使用方法详解

dict的特性 dict是python中的一个可变的数据类型,用{}表示,dict的key必须是不可变的数据类型,而value的数据类型可以任意。 格式:{key:value,key...

Python 一句话生成字母表的方法

List >>> [chr(i) for i in range(97,123)] ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h',...

解决Python出现_warn_unsafe_extraction问题的方法

解决Python出现_warn_unsafe_extraction问题的方法

在Python项目中运行出现了“AttributeError: ResourceManager instance has no attribute ‘_warn_unsafe_extra...

pytorch torch.nn.AdaptiveAvgPool2d()自适应平均池化函数详解

如题:只需要给定输出特征图的大小就好,其中通道数前后不发生变化。具体如下: AdaptiveAvgPool2d CLASStorch.nn.AdaptiveAvgPool2d(outp...

pytorch获取vgg16-feature层输出的例子

实际应用时可能比较想获取VGG中间层的输出, 那么就可以如下操作: import numpy as np import torch from torchvision import m...