我知道这是numpy中最简单的任务之一,但我不明白。我有一个数组:
import numpy as np
u1 = np.array([[[-2.04853845, -1.47283101, 1. ],
[-2.05009646, -1.46780913, 1. ],
[-2.05165539, -1.46278426, 1. ]],
[[-4.94165412, 7.8524744 , 1. ],
[ 0.94540647, 0.86456925, 1. ],
[-4.94916228, 7.87667525, 1. ]]])
字符串
u1是矩阵的汇总。矩阵的真实的大小为:
u1.shape
(813, 1200, 3)
型
我想得到最后一个数组(最后一个维度)的第一个和第二个元素在0和1之间的所有值,所以我得到一个包含布尔值的数组,这样我就可以索引另一个具有相同形状的数组。
我敢打赌这是一个重复的问题,但我没有找到它。谢啦,谢啦
3条答案
按热度按时间anhgbhbe1#
您正在寻找以下内容:
字符串
ppcbkaq52#
提取元素并与您的值进行比较:
字符串
然后提取您感兴趣的行:
型
l7wslrjt3#
定义以下函数,检查行的前两个元素:
字符串
然后将其应用于最后一个维度沿着的每个一维切片(我们称之为 * 行 *):
型
示例数据的结果为:
型