numpy.ndarray 实现对特定行或列取值

yipeiwu_com7年前Python基础

如下所示:

import numpy as np
 
b = [[1,2,0],
[4,5,0],
[7,8,1],
[4,0,1],
[7,11,1]
  ]
a=np.array([b]).reshape((5,3))
print(a)
c=[1,3,4]
# print(a[c])
d=np.nonzero(a[:, 2] == 0)
print(d)
print(a[d])
 
 

以上这篇numpy.ndarray 实现对特定行或列取值就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

python多重继承实例

本文实例讲述了python多重继承用法,分享给大家供大家参考。具体实现方法如下: 1.mro.py文件如下: #!/usr/bin/python # Filename:mro.py...

用pytorch的nn.Module构造简单全链接层实例

python版本3.7,用的是虚拟环境安装的pytorch,这样随便折腾,不怕影响其他的python框架 1、先定义一个类Linear,继承nn.Module import tor...

Django model 中设置联合约束和联合索引的方法

在Django model中对一张表的几个字段进行联合约束和联合索引,例如在购物车表中,登录的用户和商品两个字段在一起表示唯一记录。 举个栗子: Django model中购物车表...

pandas DataFrame 根据多列的值做判断,生成新的列值实例

pandas DataFrame 根据多列的值做判断,生成新的列值实例

环境:Python3.6.4 + pandas 0.22 主要是DataFrame.apply函数的应用,如果设置axis参数为1则每次函数每次会取出DataFrame的一行来做处理,如...

Python基础之文件读取的讲解

with open(filename) as fp: dataMat = [] for line in fp.readlines(): # fp.rea...