取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中的super用法详解

一、问题的发现与提出 在Python类的方法(method)中,要调用父类的某个方法,在Python 2.2以前,通常的写法如代码段1: 代码段1: 复制代码 代码如下:  c...

Python画图高斯分布的示例

如下所示: import matplotlib.pyplot as plt import numpy as np import math def gaussian(sigma, x,...

Pytorch释放显存占用方式

如果在python内调用pytorch有可能显存和GPU占用不会被自动释放,此时需要加入如下代码 torch.cuda.empty_cache() 我们来看一下官方文档的说明 Relea...

Pytorch Tensor基本数学运算详解

1. 加法运算 示例代码: import torch # 这两个Tensor加减乘除会对b自动进行Broadcasting a = torch.rand(3, 4) b = to...

python创建与遍历List二维列表的方法

python 创建List二维列表 lists = [[] for i in range(3)] # 创建的是多行三列的二维列表 for i in range(3): lists...