python - 如何用数组干净地索引 numpy 数组(或任何其他支持加法的东西，以便它可以被抵消)

<分区>

解释我的问题的最简单方法可能是举个例子，所以让我定义一些数组:

>>> test = arange(25).reshape((5,5))
>>> test
array([[ 0,  1,  2,  3,  4],
      [ 5,  6,  7,  8,  9],
      [10, 11, 12, 13, 14],
      [15, 16, 17, 18, 19],
      [20, 21, 22, 23, 24]])
>>> Xinds = array([1,2,3])
>>> Yinds = array([1,2,3])

现在，如果我想要第 1、2、3 行和第 0 列中的元素，我可以:

>>> test[Yinds,0]
array([ 5, 10, 15])

如果我想要第 1、2、3 行和所有列中的项目，我可以:

>>> test[Yinds, :]
array([[ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19]])

但是，如果我尝试扩展它以获取第 1、2 和 3 行以及第 1、2 和 3 列中的元素，- 惊喜! -- 我改为获取 (1,1)、(2,2) 和 (3,3) 中的元素

>>> test[Yinds, Xinds]
array([ 6, 12, 18])

而不是我想要的:

>>> test[Yinds, :][:, Xinds]
array([[ 6,  7,  8],
       [11, 12, 13],
       [16, 17, 18]])
>>> test[1:4,1:4]
array([[ 6,  7,  8],
       [11, 12, 13],
       [16, 17, 18]])

我意识到我可以定义一个切片，但我希望能够为索引添加一个偏移量(例如 Yinds+offset)，而这不能用切片来完成。

我可以做类似的事情

>>> xStart = 1
>>> xEnd   = 4
>>> yStart = 1
>>> yEnd   = 4

>>> offset = 1
>>> test[yStart+offset:yEnd+offset, xStart+offset:xEnd+offset]
...

或

>>> Xinds = array([1,4])
>>> Yinds = array([1,4])

>>> offset = 1
>>> test[slice(*(Yinds+offset)), slice(*(Xinds+offset))]
...

但两者都不是特别干净。

Monkey 将加法运算符修补到 slice 中似乎不是一个选项，并且从 slice 继承以添加运算符似乎也不起作用；我收到错误消息，“类型‘slice’不是可接受的基类型”。 (*Grumble* 这在 Ruby 中不是问题 *Grumble*)

所以，我的问题是，访问(多于 1 维的)子数组的最干净的方法是什么可以存储和偏移？

到目前为止的选项:

测试[Yinds+offset, :][:, Xinds+offset]
测试[yStart+offset:yEnd+offset, xStart+offset:xEnd+offset]
测试[slice(*(Yinds+offset)), slice(*(Xinds+offset))]

最佳答案

我不完全确定你想要什么，但也许 ix_有助于？我想我见过比我在类似情况下使用它更了解 numpy 的人。

>>> from numpy import array, arange, ix_
>>> a = arange(25).reshape(5,5)
>>> Xinds = array([1,2,3])
>>> Yinds = array([1,2,3])
>>> a
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24]])
>>> a[ix_(Xinds, Yinds)]
array([[ 6,  7,  8],
       [11, 12, 13],
       [16, 17, 18]])
>>> a[ix_(Xinds+1, Yinds)]
array([[11, 12, 13],
       [16, 17, 18],
       [21, 22, 23]])
>>> Y2inds = array([1,3,4])
>>> a[ix_(Xinds, Y2inds)]
array([[ 6,  8,  9],
       [11, 13, 14],
       [16, 18, 19]])
>>> a[ix_(Xinds, Y2inds-1)]
array([[ 5,  7,  8],
       [10, 12, 13],
       [15, 17, 18]])

关于python - 如何用数组干净地索引 numpy 数组(或任何其他支持加法的东西，以便它可以被抵消)，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/10989438/

上一篇：python - 如何检查线程当前是否持有 GIL？

下一篇：Python 对象持久化

相关文章：

python - 取消嵌套嵌套列表

python - 数组列表 : How can I see if an array in one list is in another list?

python - 没有使用 gevent 加速读取文件

python - 大型 Numpy Scipy CSR 矩阵，按行操作

numpy - numpy中的frombuffer和fromiter有什么区别？为什么以及何时使用这些

mysql - 文档词索引数据库结构？

python - 将文本添加到文件的第一行

mysql - 在 MySQL 中，将要求和的两列转换为索引有好处吗？

oracle - 在 Oracle 中为大表的新列创建索引

通过 tp_getset 访问时 Python C 扩展出现段错误