小编典典

这种numpy选择行为的背后是什么?

python

在回答这个问题时,我和其他一些人实际上认为是错误的,因为认为以下方法可行:

说一个有

test = [ [ [0], 1 ],
         [ [1], 1 ]
       ]
import numpy as np
nptest = np.array(test)

背后的原因是什么

>>> nptest[:,0]==[1]
array([False, False], dtype=bool)

而有一个

>>> nptest[0,0]==[1],nptest[1,0]==[1]
(False, True)

要么

>>> nptest==[1]
array([[False,  True],
       [False,  True]], dtype=bool)

要么

>>> nptest==1
array([[False,  True],
       [False,  True]], dtype=bool)

是造成尺寸 退化的 原因吗?


阅读 193

收藏
2021-01-20

共1个答案

小编典典

nptest 是对象dtype的2D数组,每行的第一个元素是一个列表。

nptest[:, 0] 是对象dtype的一维数组,其每个元素都是列表。

当您这样做时nptest[:,0]==[1],NumPy不会nptest[:,0]对list的每个元素进行元素比较[1]。它从中创建尽可能高维的数组[1],生成1D数组np.array([1]),然后广播比较,将的每个元素nptest[:,0]与整数1进行比较。

由于innptest[:, 0]中的列表都不等于1,因此结果的所有元素均为False。

2021-01-20