Numpy 子集化和赋值

Question

假设您想要将 NumPy 数组子集fridge_items化为评分高于 7 的美味食物行。

most_tasty_items = fridge_items[:,10] > 7)

你得到一个布尔值数组。

如果你当时做了：

fridge_items[most_tasty_items,:][:3,:]

当您索引到 fridge_items 时，这里到底发生了什么。我熟悉做 array[1,2] 并返回给定行和列的内容。

由于most_tasty_items是布尔值的一维数组，我们如何使用[:3,:]? 如果它只是一个一维数组，我们可以直接说 [:]。不太明白这个，为什么我们把:第二个参数给[most_tasty_items,:]

Answer 1

当您使用两对括号寻址数据时，您将执行两个操作，第一个括号从数据中选择一个新数组，第二个括号寻址新数组。In [71]: np.random.seed(2020)     ...: fridge = np.random.randint(11, size=(30, 5))     ...: tasty = fridge_items[:,4] > 7     ...: tastyfridge = fridge[tasty,:]                                                    In [72]: tastyfridge[:2,:], fridge[tasty][:2,:]                                           Out[72]: (array([[ 8, 10,  9,  3,  7],        [ 4,  7,  1,  4,  9]]), array([[ 8, 10,  9,  3,  7],        [ 4,  7,  1,  4,  9]]))In [73]: