程序员经验分享-hwhale

使用 2 个索引列表对 2D Numpy 数组进行索引

2023-12-20

我遇到了一个奇怪的情况。

我有一个 2D Numpy 数组 x：

x = np.random.random_integers(0,5,(20,8))

我有 2 个索引器——一个带有行索引，另一个带有列索引。为了索引 X，我必须执行以下操作：

row_indices = [4,2,18,16,7,19,4]
col_indices = [1,2]
x_rows = x[row_indices,:]
x_indexed = x_rows[:,column_indices]

而不仅仅是：

x_new = x[row_indices,column_indices]

（失败并显示：错误，无法使用 (2,) 广播 (20,)）

我希望能够使用广播在一行中进行索引，因为这将使代码保持干净和可读...另外，我对 python 的底层了解不多，但据我了解它，在一行中完成它应该更快（并且我将使用相当大的数组）。

测试用例：

x = np.random.random_integers(0,5,(20,8))

row_indices = [4,2,18,16,7,19,4]
col_indices = [1,2]
x_rows = x[row_indices,:]
x_indexed = x_rows[:,col_indices]

x_doesnt_work = x[row_indices,col_indices]

选择或作业np.ix_使用索引或布尔数组/掩码

1. With indexing-arrays

一个选择

我们可以用np.ix_ https://docs.scipy.org/doc/numpy/reference/generated/numpy.ix_.html获取可相互广播的索引数组元组，以产生更高维的索引组合。因此，当该元组用于索引输入数组时，将为我们提供相同的高维数组。因此，要根据两个因素进行选择1D索引数组，那就是 -

x_indexed = x[np.ix_(row_indices,col_indices)]

B、作业

我们可以使用相同的符号将标量或可广播数组分配到这些索引位置。因此，以下作业适用于 -

x[np.ix_(row_indices,col_indices)] = # scalar or broadcastable array

2. With masks

我们还可以使用布尔数组/掩码np.ix_，类似于索引数组的使用方式。这可以再次用于从输入数组中选择一个块，也可以对其进行赋值。

一个选择

因此，随着row_mask and col_mask布尔数组分别作为行和列选择的掩码，我们可以使用以下内容进行选择 -

x[np.ix_(row_mask,col_mask)]

B、作业

以下适用于作业 -

x[np.ix_(row_mask,col_mask)] = # scalar or broadcastable array

样品运行

1. 使用np.ix_ with indexing-arrays

输入数组和索引数组 -

In [221]: x
Out[221]: 
array([[17, 39, 88, 14, 73, 58, 17, 78],
       [88, 92, 46, 67, 44, 81, 17, 67],
       [31, 70, 47, 90, 52, 15, 24, 22],
       [19, 59, 98, 19, 52, 95, 88, 65],
       [85, 76, 56, 72, 43, 79, 53, 37],
       [74, 46, 95, 27, 81, 97, 93, 69],
       [49, 46, 12, 83, 15, 63, 20, 79]])

In [222]: row_indices
Out[222]: [4, 2, 5, 4, 1]

In [223]: col_indices
Out[223]: [1, 2]

索引数组的元组np.ix_ -

In [224]: np.ix_(row_indices,col_indices) # Broadcasting of indices
Out[224]: 
(array([[4],
        [2],
        [5],
        [4],
        [1]]), array([[1, 2]]))

做出选择 -

In [225]: x[np.ix_(row_indices,col_indices)]
Out[225]: 
array([[76, 56],
       [70, 47],
       [46, 95],
       [76, 56],
       [92, 46]])

As OP建议 https://stackoverflow.com/questions/35607818/index-a-2d-numpy-array-with-2-lists-of-indices#comment58900519_35608029，这实际上与执行相同老套使用 2D 数组版本进行广播row_indices有它的元素/索引sent to axis=0从而创建一个单例维度axis=1从而允许广播col_indices。因此，我们会有一个替代解决方案，如下所示 -

In [227]: x[np.asarray(row_indices)[:,None],col_indices]
Out[227]: 
array([[76, 56],
       [70, 47],
       [46, 95],
       [76, 56],
       [92, 46]])

正如前面所讨论的，对于作业，我们只是这样做。

行、列索引数组 -

In [36]: row_indices = [1, 4]

In [37]: col_indices = [1, 3]

使用标量进行赋值 -

In [38]: x[np.ix_(row_indices,col_indices)] = -1

In [39]: x
Out[39]: 
array([[17, 39, 88, 14, 73, 58, 17, 78],
       [88, -1, 46, -1, 44, 81, 17, 67],
       [31, 70, 47, 90, 52, 15, 24, 22],
       [19, 59, 98, 19, 52, 95, 88, 65],
       [85, -1, 56, -1, 43, 79, 53, 37],
       [74, 46, 95, 27, 81, 97, 93, 69],
       [49, 46, 12, 83, 15, 63, 20, 79]])

使用 2D 块（可广播数组）进行分配 -

In [40]: rand_arr = -np.arange(4).reshape(2,2)

In [41]: x[np.ix_(row_indices,col_indices)] = rand_arr

In [42]: x
Out[42]: 
array([[17, 39, 88, 14, 73, 58, 17, 78],
       [88,  0, 46, -1, 44, 81, 17, 67],
       [31, 70, 47, 90, 52, 15, 24, 22],
       [19, 59, 98, 19, 52, 95, 88, 65],
       [85, -2, 56, -3, 43, 79, 53, 37],
       [74, 46, 95, 27, 81, 97, 93, 69],
       [49, 46, 12, 83, 15, 63, 20, 79]])

2. 使用np.ix_ with masks

输入数组 -

In [19]: x
Out[19]: 
array([[17, 39, 88, 14, 73, 58, 17, 78],
       [88, 92, 46, 67, 44, 81, 17, 67],
       [31, 70, 47, 90, 52, 15, 24, 22],
       [19, 59, 98, 19, 52, 95, 88, 65],
       [85, 76, 56, 72, 43, 79, 53, 37],
       [74, 46, 95, 27, 81, 97, 93, 69],
       [49, 46, 12, 83, 15, 63, 20, 79]])

输入行、列掩码 -

In [20]: row_mask = np.array([0,1,1,0,0,1,0],dtype=bool)

In [21]: col_mask = np.array([1,0,1,0,1,1,0,0],dtype=bool)

做出选择 -

In [22]: x[np.ix_(row_mask,col_mask)]
Out[22]: 
array([[88, 46, 44, 81],
       [31, 47, 52, 15],
       [74, 95, 81, 97]])

使用标量进行赋值 -

In [23]: x[np.ix_(row_mask,col_mask)] = -1

In [24]: x
Out[24]: 
array([[17, 39, 88, 14, 73, 58, 17, 78],
       [-1, 92, -1, 67, -1, -1, 17, 67],
       [-1, 70, -1, 90, -1, -1, 24, 22],
       [19, 59, 98, 19, 52, 95, 88, 65],
       [85, 76, 56, 72, 43, 79, 53, 37],
       [-1, 46, -1, 27, -1, -1, 93, 69],
       [49, 46, 12, 83, 15, 63, 20, 79]])

使用 2D 块（可广播数组）进行分配 -

In [25]: rand_arr = -np.arange(12).reshape(3,4)

In [26]: x[np.ix_(row_mask,col_mask)] = rand_arr

In [27]: x
Out[27]: 
array([[ 17,  39,  88,  14,  73,  58,  17,  78],
       [  0,  92,  -1,  67,  -2,  -3,  17,  67],
       [ -4,  70,  -5,  90,  -6,  -7,  24,  22],
       [ 19,  59,  98,  19,  52,  95,  88,  65],
       [ 85,  76,  56,  72,  43,  79,  53,  37],
       [ -8,  46,  -9,  27, -10, -11,  93,  69],
       [ 49,  46,  12,  83,  15,  63,  20,  79]])
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)

python

Arrays

NumPy

使用 2 个索引列表对 2D Numpy 数组进行索引 的相关文章

随机推荐

  • iPhone：禁用“双击空格键。”捷径？

    默认情况下 如果您在 iPhone 或 iPad 上点击空格键两次 则不会得到 两个空格 而是得到 句点后跟一个空格 有什么方法可以在代码中禁用此快捷方式吗 更新 通过 UITextInputTraits 禁用自动更正不起作用 更新2 明白

  • Scala中涉及抽象类型时没有动态绑定？

    当我在 Martin Odersky 的抽象类型中尝试动物 食品示例时Scala 编程 class Food abstract class Animal type SuitableFood lt Food def eat food Suit

  • 将 vue-router 组件解释为一个函数

    我在几个不同的地方看到了以下类型的路由定义 path dashboard component render c return c router view children path component Dashboard 我试图理解这有什么

  • Linux 中的 SDL 桌面分辨率检测[重复]

    这个问题在这里已经有答案了 我收到一些报告称 对于某些 Linux 用户 尤其是使用 SteamOS 的用户 我的游戏以错误的分辨率打开 游戏尝试检测当前桌面分辨率并使用该分辨率创建无边框全屏窗口 例如 SteamOS 的分辨率通常为 19

  • java.lang.StackOverflowError：在 ListView 适配器中启用/禁用 TextView#setTextIsSelectable 时堆栈大小为 8MB

    我有一个活动ListView 显示TextView在每个列表项中 使用这些方法切换选定位置的属性EditText setTextIsSelectable EditText setEllipsize EditText setSingleLin

  • XCTest相对于SenTestingKit有哪些优势？

    我想在 iOS 中进行单元测试 默认情况下 测试类是使用 Xcode 5 中的 XCTest 框架类创建的 我在早期版本的 iOS 中使用了 SenTestingKit 这两个框架有什么区别 XCTest框架有哪些优势 我用谷歌搜索了相关文

  • 材料表使行在单击时可编辑

    使用材料表 https material table com 库中 我试图使表行可以通过双击进行编辑 单击该行应该与单击操作列最左侧的编辑按钮具有相同的效果 我已成功链接到正确的事件处理程序 现在双击一行时由警报框表示 https code

  • kohana3.0如何彻底删除镜像文件？

    我有一个 kohana3 0 内容和文件管理系统 当用户删除图像时 我希望能够从数据库以及我上传图像的文件夹中完全删除图像 现在我用于图像删除 public function delete id NULL parent delete id

  • gdb - 执行命令n次

    我正在使用 gdb 调试器来运行一个包含循环的程序 让我们坐在 10 中 当我进入循环时 我不想进入它 我知道我可以在循环之后设置第二个断点 而不是 c 继续 但我也记得有可能做类似 n 10 接下来的 10 次 的事情 n 10 不起作用

  • 如何使用 Ansible 创建新分区

    当我在命令行上运行它时 它工作正常 echo e n np n1 n n nw sudo fdisk dev sdb 但在 Ansible 中它不想在 shell 中运行 name partition new disk shell echo

  • 如何在运行时填充 boost::fusion::vector ？

    首先 为与我之前的问题相似而道歉here https stackoverflow com questions 13094535 iterating over boost fusionvector 但我认为我问得不对 我有一个方法 templ

  • 在 .net 2.0 应用程序中加载 .net 3.5 wpf-forms

    我正在尝试在 net 2 0 Windows 窗体应用程序中加载和托管 WPF 控件 仅当安装了 net 3 5 时才应加载 WPF 控件 我找到了一个链接在 MFC 应用程序中托管 WPF 内容 http www codeguru com

  • npm 认为 Node 已经过时了，但事实并非如此

    我正在尝试安装node gyp 我跑 sudo npm install g node gyp 我收到一条错误 提示它需要节点 0 8 0 并提示我正在运行 0 6 12 问题是 我不是 我正在使用 NVM 目前正在运行 0 11 somet

  • 如何将 numpy 数组列表转换为单个 numpy 数组？

    假设我有 LIST array 1 2 3 4 5 array 1 2 3 4 5 1 2 3 4 5 inner lists are numpy arrays 我尝试转变 array 1 2 3 4 5 1 2 3 4 5 1 2 3 4

  • 无法在 Google App Engine 上运行 Flask 调试模式

    我正在使用 Python 2 7 在 Google App Engine 上运行 Flask 0 9 Werkzeug 0 8 3 并且我非常希望运行 Werkzeug 调试器 尝试使用后werkzeug appengine debugge

  • 未知管理员：ComponentInfo - 尝试为生产 Android 设备启用 kiosk 模式？

    我正在尝试为我的设备启用信息亭模式 但收到未知的管理错误 我很确定我成功地将我的应用程序设置为设备所有者 并且我认为这是最后一步 但这个错误给我带来了很多麻烦 清单 我这里有一个接收器 应该将我的应用程序绑定为设备管理员 它还引用了 dev

  • 在javascript中使用window.open()有什么限制吗

    使用有限制吗window open URL 在 JavaScript 中 当我尝试使用打开三个窗口时window open URL 打开第二个窗口后 第三个窗口没有单独打开 但它只是刷新第一个窗口的内容并在其中显示第三个窗口的内容 使用 b

  • Java中showMessageDialog中的frame和null有什么区别？

    假设我有一个 JFrame 类的对象作为框架 我想知道有什么区别 JOptionPane showMessageDialog null message and JOptionPane showMessageDialog frame mess

  • iOS：如何修剪 .aif 音频录制开始和结束时的静音？

    我的应用程序包括用户录制简短消息的功能 我想从录音的开头和结尾删除任何静音 或者更准确地说 任何音量低于给定阈值的音频 我正在使用 AVAudioRecorder 录制音频 并将其保存到 aif 文件中 我在其他地方看到过一些提到的方法 通

  • 使用 2 个索引列表对 2D Numpy 数组进行索引

    我遇到了一个奇怪的情况 我有一个 2D Numpy 数组 x x np random random integers 0 5 20 8 我有 2 个索引器 一个带有行索引 另一个带有列索引 为了索引 X 我必须执行以下操作 row indi

热门标签