1
是大于阈值元素的索引我有一个二维数组numpy的:获取在2D numpy的阵列
x = [[ 1.92043482e-04 0.00000000e+00 0.00000000e+00 0.00000000e+00
0.00000000e+00 0.00000000e+00 2.41005634e-03 0.00000000e+00
7.19330120e-04 0.00000000e+00 0.00000000e+00 1.42886875e-04
0.00000000e+00 0.00000000e+00 0.00000000e+00 0.00000000e+00
0.00000000e+00 9.79279411e-05 7.88888657e-04 0.00000000e+00
0.00000000e+00 1.40425916e-01 0.00000000e+00 1.13955893e-02
7.36868947e-03 3.67091988e-04 0.00000000e+00 0.00000000e+00
0.00000000e+00 0.00000000e+00 1.72037105e-03 1.72377961e-03
0.00000000e+00 0.00000000e+00 1.19532061e-01 0.00000000e+00
0.00000000e+00 0.00000000e+00 0.00000000e+00 3.37249481e-04
0.00000000e+00 0.00000000e+00 0.00000000e+00 0.00000000e+00
0.00000000e+00 0.00000000e+00 1.75111492e-03 0.00000000e+00
0.00000000e+00 1.12639313e-02]
[ 0.00000000e+00 0.00000000e+00 1.10271735e-04 5.98736562e-04
6.77961628e-04 7.49569659e-04 0.00000000e+00 0.00000000e+00
2.91697850e-03 0.00000000e+00 0.00000000e+00 0.00000000e+00
0.00000000e+00 0.00000000e+00 3.30257021e-04 2.46629275e-04
0.00000000e+00 1.87586441e-02 6.49103144e-04 0.00000000e+00
1.19046355e-04 0.00000000e+00 0.00000000e+00 2.69499898e-03
1.48525386e-02 0.00000000e+00 0.00000000e+00 0.00000000e+00
0.00000000e+00 0.00000000e+00 0.00000000e+00 1.18803119e-03
3.93100829e-04 0.00000000e+00 3.76245304e-04 2.79537738e-02
0.00000000e+00 1.20738457e-03 9.74669064e-06 7.18680093e-04
1.61546793e-02 3.49360861e-04 0.00000000e+00 0.00000000e+00
0.00000000e+00 0.00000000e+00 0.00000000e+00 0.00000000e+00
0.00000000e+00 0.00000000e+00]]
我如何获得了比0.01
更大的元素的索引?
现在,我正在做t = np.argmax(x, axis=1)
以获得每个最大值的索引,结果是:[21 35]
。我如何实现上述目标?
嗯,我想我不是如何解释NP阵列清楚。 In(array([0,1]),array([1,0])),为什么是数组[1,0]而不是[0,1]? – Arman
,因为他们是numpy对象 – maxymoo
我不确定这是否给了我正确的结果。我做了t = np.where(x> 0.01,我得到的输出是:(array([0,0,0,0,1,1,1,1]),array([21,23,34, 49,17,24,35,40]))。但是,第一个元素甚至没有21个元素,当使用argmax返回时,返回值为 – Arman