更新python函数中的引用变量

Question

我正在尝试通过类 P 更改矩阵 X 的顺序，该类 P 已经传递给基类 B 中的函数。顺序的更改不会反映在基类 B 中。有没有办法实现这个？请参阅以下 MWE：

#p.py
import numpy as np
import b

class P(b.B):

    def __init__(self, opts):
        self.opts = opts

    def func1(self, X):

        print('func1:',X)

    def func2(self):

        randomize = np.arange(len(self.myX))
        np.random.shuffle(randomize)
        self.myX = self.myX[randomize]

        print('func2',self.myX)

    def func3(self, X):
        """
        X is a 2d matrix 
        """
        self.myX = X
        #call the function from the base class 
        b.B.func3(self, self.myX)  

#b.py
class B:
    """
    base class.

    """
    def __init__(self, opts):
        self.opts = opts


    def func3(self, X):

        for n in range(2):
            for i in range(X.shape[0]):
                self.func1(X[i])
            self.func2()

从控制台：

p1 = p.P({})
X=np.array([[1,2,3], [2,3,4], [3,4,5], [4,5,6], [5,6,7], [6,7,8]])
p1.func3(X)

电流输出：

func1: [1 2 3]
func1: [2 3 4]
func1: [3 4 5]
func1: [4 5 6]
func1: [5 6 7]
func1: [6 7 8]
func2 [[6 7 8]
 [3 4 5]
 [2 3 4]
 [5 6 7]
 [4 5 6]
 [1 2 3]]
func1: [1 2 3]
func1: [2 3 4]
func1: [3 4 5]
func1: [4 5 6]
func1: [5 6 7]
func1: [6 7 8]

预期输出：

func1: [1 2 3]
func1: [2 3 4]
func1: [3 4 5]
func1: [4 5 6]
func1: [5 6 7]
func1: [6 7 8]
func2 [[6 7 8]
 [3 4 5]
 [2 3 4]
 [5 6 7]
 [4 5 6]
 [1 2 3]]
func1: [6 7 8]
func1: [3 4 5]
func1: [2 3 4]
func1: [5 6 7]
func1: [4 5 6]
func1: [1 2 3]

所以基本上，当控件在B中从p.func2返回到func3时，X应该和self.myX一样。我相信这应该发生，因为 self.myX 默认通过引用传递给 b.func3。

Answer 1

问题出在这里：

#class p.P
    def func2(self):

        randomize = np.arange(len(self.myX))
        np.random.shuffle(randomize)
        self.myX = self.myX[randomize]  # <-- this line

        print('func2',self.myX)

您有效地将self.myX 重新分配给一个新 随机np 数组，将对象引用从X 更改为这个新的随机数组。

如果您在此行之间添加了print(id(self.myX))，您会注意到最初的参考点又回到了X，但是一旦您重新分配它，id 就不再一样了。

您应该这样做以维护参考：

np.random.shuffle(self.myX)

如果你真的也想打乱内部数组：

for arr in self.myX:
    np.random.shuffle(arr)
np.random.shuffle(self.myX)

这也将保持引用。

编辑：如果您希望在保持对象的同时保留随机顺序的引用，这有点棘手但可行：

# use make randomize an instance attribute so you can refer to it even after the function ended
self.shuffle_order = np.arange(len(self.myX))
np.random.shuffle(self.shuffle_order)

# loop through each inner array and reassign based on a copy of the shuffled matrix
for i, arr in enumerate(self.myX[self.shuffle_order]):
    self.myX[i] = arr

这会更改内部数组，但会保持self.myX 和X 之间的整体对象引用。

之后您可以通过p1.shuffle_order检索订单。