使用库numpy

Matlab中

通过逻辑矩阵可快速得到

a = [1,2,3,4,5,6]a =1   2   3   4   5   6b = logical([1,0,0,1,1,0])b =1  0  0  1  1  0a(b)ans =1   4   5

为了在python中达到相同效果！！

Python中

导入numpy

import numpy as np

创建一个23的int型矩阵和一个23的bool型矩阵

a = np.array([[1,2,3],[4,5,6]])
b = np.array([[True,True,False],[False,True,False]])

矩阵直观显示

[[1 2 3][4 5 6]]
[[ True  True False][False  True False]]

a, b两矩阵相乘

c = a*b

结果

[[1 2 0][0 5 0]]

numpy.flatnonzero可返回a的展平版本中非零的索引。

np.flatnonzero(c)

结果

array([0, 1, 4], dtype=int64)

使用非零元素的索引作为索引数组来提取这些元素

c.ravel()[np.flatnonzero(c)]

结果

[1 2 5]

完整过程

import numpy as npa = np.array([[1,2,3],[4,5,6]])
b = np.array([[True,True,False],[False,True,False]])
c = a*b
print(c.ravel()[np.flatnonzero(c)])out:
[1 2 5]

python库numpy使用技巧（一）——提取数组中非零元素相关推荐

python库numpy使用技巧（二）——随机抽取二维矩阵中多行或多列
使用库numpy 创建一个二维数组 import numpy as nparray = np.arange(24).reshape((4,6))""" array([[ ...
python数据分析获取数组中非零元素的索引
[小白从小学Python.C.Java] [Python-计算机等级考试二级] [Python-数据分析] python数据分析获取数组中非零元素的索引选择题以下说法不正确的是? import ...
返回数组中非零元素的索引np.flatnonzero()
[小白从小学Python.C.Java] [计算机等考+500强证书+考研] [Python-数据分析] 返回数组中非零元素的索引 np.flatnonzero() 选择题以下程序的运行结果是? i ...
【Python】获取数组中非零元素
第一种方法flatnonzero import numpy as npa = np.array([[1,2,3],[4,5,6]]) b = np.array([[True,True,False],[ ...
python库numpy基本用法
NumPy(Numerical Python) 是 Python 语言的一个扩展程序库,支持大量的维度数组与矩阵运算,此外也针对数组运算提供大量的数学函数库. 1. 数组创建 import numpy ...
centos7 python3安装numpy_CentOS7 安装python库(numpy、scipy、matplotlib、scikit-learn、tensorflow)...
0.1准备工作安装好CentOS7,配置好网络,确保网络畅通. 0.2root授权首先:当前用户为kaid # vim /etc/sudoers 在root ALL=(ALL) ALL之后添加: ...
python库--Numpy and pandas
list转为矩阵 array = numpy.array([[1,2,3] [4,5,6]] ) 矩阵的维度属性:array.ndim # 注意这是最小的那个 array.shape # 几行几列 a ...
python库numpy的reshape的终极解释
a = np.arange(2*4*4) b = a.reshape(1,4,4,2) #应该这样按反序来理解:最后一个2是一个只有2个元素的向量,最后的4,2代表4×2的矩阵,最 ...
python 矩阵元素查找位置函数_Python中二维数组中非零元素位置的快速查找方法...
我提出这个问题的原因是我认为numpy.非零功能没有得到应有的优化.以下示例显示了这一事实:a=np.random.random((1000,1000)) a[a<0.5]=0 timeit.t ...

python库numpy使用技巧（一）——提取数组中非零元素

Matlab中

Python中

完整过程

python库numpy使用技巧（一）——提取数组中非零元素相关推荐

最新文章

热门文章