2

我正在尝试将一些 Matlab 代码翻译成 Python（使用 NumPy）。我对 Matlab 不是很熟悉，并且遇到了无法解析的一行：

w(idx(1:p, 1), 1) = v(idx(1:p, 1), 1) - theta;

我冒昧地猜测一个p-long headx被用作选择p的条目的索引w，并且 in 的那些条目w被替换为 in 的相应条目v（少一个 scalar theta）。

在 Octave 中四处乱窜，这似乎是对它正在做什么的准确描述，但我找不到任何关于这种效果的文档。

无论如何，用 Python 重写这段代码的最佳方法是什么？我查看了 NumPy 的“暂定教程”，试图找到一种优雅的方式来完成它，看起来这可能就是我正在寻找的。但是，我很难让它看起来不错，尤其是使用赋值运算符。是否有任何更优雅或 Python 惯用的方法来执行此分配操作？

4

2 回答 2

1

这基本上是@Dan 在评论中写的，但考虑了 python 中从零开始的索引：

w[idx[:p, 0], 0] = v[idx[:p, 0], 0] - theta

不确定您是否想要比这更优雅的东西。如果仅应修改第一列，则需要这些零。

于 2013-05-06T15:48:44.620 回答
0

你对基本行为是正确的。索引矩阵第一列的一个p长度子向量idx被用于选择元素v并将它们放在矩阵中的相同位置，w然后首先通过标量调整它们的值theta。

对 MATLAB 使用从一开始的索引，对numpy.

在 MATLAB 中，

clear

% Data matrices
w = zeros(5,5)
v = diag([10,20,30,40,50]) * ones(5,5)

% Indexing matrix
idx = ceil(5*rand(5, 5))

% Selection and adjustment parameters
p = 3    
theta = 1

% Apply adjustment and selection
w(idx(1:p, 1), 1) = v(idx(1:p, 1), 1) - theta

产生输出

w =

     0     0     0     0     0
     0     0     0     0     0
     0     0     0     0     0
     0     0     0     0     0
     0     0     0     0     0


v =

    10    10    10    10    10
    20    20    20    20    20
    30    30    30    30    30
    40    40    40    40    40
    50    50    50    50    50


idx =

     3     1     2     3     4
     1     1     2     1     3
     4     1     2     2     2
     1     1     5     1     1
     1     2     4     5     4


theta =

     1


p =

     3


w =

     9     0     0     0     0
     0     0     0     0     0
    29     0     0     0     0
    39     0     0     0     0
     0     0     0     0     0

并且，等效的 Python 代码使用numpy

import numpy as np

# Data arrays
w = np.zeros((5,5))
v = np.dot(np.diag([10, 20, 30, 40, 50]), np.ones((5,5)))
print "w = "
print w
print "v = "
print v

# Indexing array
idx = np.floor(5 * np.random.rand(5,5)).astype(int)
print "idx = "
print idx

# Selection and adjustment parameters
theta = 1
p = 3

# Apply selection and adjustment
w[idx[:p, 0], 0] = v[idx[:p, 0], 0] - theta
print "w = "
print w

产生输出

w = 
[[ 0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.]]
v = 
[[ 10.  10.  10.  10.  10.]
 [ 20.  20.  20.  20.  20.]
 [ 30.  30.  30.  30.  30.]
 [ 40.  40.  40.  40.  40.]
 [ 50.  50.  50.  50.  50.]]
idx = 
[[0 2 2 0 3]
 [1 2 1 2 4]
 [2 2 4 3 4]
 [0 1 1 4 4]
 [0 1 0 4 3]]
w = 
[[  9.   0.   0.   0.   0.]
 [ 19.   0.   0.   0.   0.]
 [ 29.   0.   0.   0.   0.]
 [  0.   0.   0.   0.   0.]
 [  0.   0.   0.   0.   0.]]
于 2013-05-06T16:08:50.113 回答