从MATLAB到Python的联盟转换

Question

我有一个二维矩阵 C 在Python中，我试图做在MATLAB中做的事情。

MATLAB代码。

Tr = 5;
Td = 4;
Gr = 4;
Gd = 1;
C(union(1:(Tr+Gr),end-(Tr+Gr-1):end), :) = 0;
C(:, union(1:(Td+Gd),end-(Td+Gd-1):end)) = 0;

Python代码：

ru1 = C[:Tr+Gr]
a = np.shape(C)
ru2 = C[a[0] - (Tr + Gr -1):]
C[ru1.union(ru2),:]= 0

然而，我得到一个错误 union. 我怎样才能取这些数组的联合？

Answer 1

试试 numpy.union1d. union() 只适用于集合（用 {} )