numpy:矩阵的元素选取

简介: numpy:矩阵的元素选取
>>> a = np.random.random((3,4))
>>> a
array([[0.890582  , 0.43671675, 0.19531267, 0.13555252],
       [0.17254557, 0.3088476 , 0.36916878, 0.81551741],
       [0.31869139, 0.16168486, 0.58004271, 0.4279148 ]])
>>> a.nonzero()
(array([0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2]), array([0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3]))
>>> a[a.nonzero()]   ##方法一,使用矩阵去选择
array([0.890582  , 0.43671675, 0.19531267, 0.13555252, 0.17254557,
       0.3088476 , 0.36916878, 0.81551741, 0.31869139, 0.16168486,
       0.58004271, 0.4279148 ])
>>> a
array([[0.890582  , 0.43671675, 0.19531267, 0.13555252],
       [0.17254557, 0.3088476 , 0.36916878, 0.81551741],
       [0.31869139, 0.16168486, 0.58004271, 0.4279148 ]])
>>> a[1,1]         ##方法二
0.3088475978138986
>>> a[1][1]         ##方法三
0.3088475978138986
>>> a[np.array([[False,False,False,False],[True,True,True,True,],
    [False,False,False,False]])]
array([0.72159605, 0.90488492, 0.60745105, 0.20538722])  ##方法四
目录
相关文章
|
1月前
|
存储 机器学习/深度学习 算法
【算法训练-数组 三】【数组矩阵】螺旋矩阵、旋转图像、搜索二维矩阵
【算法训练-数组 三】【数组矩阵】螺旋矩阵、旋转图像、搜索二维矩阵
40 0
|
1月前
|
机器学习/深度学习 存储 人工智能
利用前缀和计算二维矩阵子矩阵的和
利用前缀和计算二维矩阵子矩阵的和
33 0
|
1月前
|
算法
【Leetcode 74】搜索二维矩阵 —— 二分查找|矩阵
给你一个满足下述两条属性的`m x n`整数矩阵:每行中的整数从左到右按非严格递增顺序排列。每行的第一个整数大于前一行的最后一个整数
|
11月前
|
数据挖掘
kmeans聚类质心个数选取的10种方式
kmeans聚类质心个数选取的10种方式
100 0
torch, 如何生成主对角阵,对角元素要么为0,要么为1
除了使用 torch.eye() 函数生成主对角阵之外,还可以使用 torch.diag() 函数来生成主对角阵。torch.diag() 函数既可以将一个向量转化为对角矩阵,也可以将一个对角矩阵中的对角元素抽取出来形成一个向量。 下面是使用 torch.diag() 函数生成主对角阵的示例代码:
660 0
|
资源调度 PyTorch 算法框架/工具
pytorch 如何生成指定位置、尺度参数的随机高斯矩阵,并指定这个随机矩阵的形式
在上述代码中,我们使用 torch.normal(mean=mu, std=sigma, size=(m, n)) 函数直接生成了一个形状为 (m, n) 的随机高斯矩阵 data,其中 mean 参数指定了均值,std 参数指定了标准差。 需要注意的是,与 torch.randn() 不同,torch.normal() 生成的是具有指定均值和标准差的高斯分布,因此生成的随机矩阵不一定是标准正态分布。如果需要生成标准正态分布随机矩阵,可以将 mean 参数设置为 0,std 参数设置为 1。
978 1
|
Python
Python经典编程习题100例:第44例:两个 3 行 3 列的矩阵,实现其对应位置的数据相加,并返回一个新矩阵
Python经典编程习题100例:第44例:两个 3 行 3 列的矩阵,实现其对应位置的数据相加,并返回一个新矩阵
263 0
|
索引
np数组的变形与合并
np数组的变形与合并
86 0
|
存储 开发者
R 数据框、因子和列表|学习笔记
向量,矩阵和数组,它们共同的特点就是它们中的数据只能拥有一组模式,也就是可以尝试字符型,也可以尝试数值类型,但是不允许在一个向量中即出现字符型也出现数值型,如果一个数据类型中即想放数值型,也想放字符型,也就是不同的模式的数据,应该用什么样的方式进行存储?数据框就可以存放不同类型的数据。
109 0
R 数据框、因子和列表|学习笔记
|
算法
LeetCode 73矩阵置零&74搜素二维矩阵&75颜色分类
给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用原地算法。
107 0
LeetCode 73矩阵置零&74搜素二维矩阵&75颜色分类