基于Numpy的类MATLAB数组索引
基于Numpy的类MATLAB数组索引
提问于 2012-07-09 18:21:40
在MATLAB和Numpy中，数组都可以通过数组进行索引。但是，行为是不同的。让我用一个例子来解释这一点。
MATLAB：
>> A = rand(5,5)
A =
0.1622 0.6020 0.4505 0.8258 0.1067
0.7943 0.2630 0.0838 0.5383 0.9619
0.3112 0.6541 0.2290 0.9961 0.0046
0.5285 0.6892 0.9133 0.0782 0.7749
0.1656 0.7482 0.1524 0.4427 0.8173
>> A([1,3,5],[1,3,5])
ans =
0.1622 0.4505 0.1067
0.3112 0.2290 0.0046
0.1656 0.1524 0.8173Numpy：
In [2]: A = arange(25).reshape((5,5))
In [3]: A
Out[3]:
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]])
In [6]: A[[0,2,4], [0,2,4]]
Out[6]: array([ 0, 12, 24])换句话说: MATLAB选择行和列，Numpy“压缩”两个索引数组，并使用元组指向条目。
如何使用Numpy获得MATLAB的行为？
Stack Overflow用户
回答已采纳
发布于 2012-07-09 18:52:05
您可以这样做：
A[[0,2,4],:][:,[0,2,4]]这将给出您想要的类似MATLAB的结果。
值得注意的是，如果你使用切片进行索引，那么你就会得到类似于MATLAB的结果，而不需要任何这样的破解：
>>> A[1:3,1:3]
array([[ 6, 7],
[11,12]])与MATLAB不同，在[1,2]中，1:3不仅仅是numpy或任何类似东西的缩写。(在这一点上，我觉得有必要提到一些您肯定已经知道的事情，即Python语言的1:3有点像[1,2]，而MATLAB的语言有点像[1,2,3]：右边的端点包含在MATLAB中，而不包括在Python中。)
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/11393140
复制相关文章
相似问题
腾讯云开发者
