Python3在给定一个完整数组和一个子数组时，如何获取rest数组？

我有一个数组和一个子数组，如下所示：

import numpy as np    
total_array = np.arange(10)
sub_array = np.array([0, 1, 2, 3])

我想知道如何获得

rest\u array=total\u array-sub\u array=np.array（[4,5,6,7,8,9]）

有什么简单的想法吗？提前感谢。

您只需使用即可获得想要的结果

import numpy as np    
total_array = np.arange(10)
sub_array = np.array([0, 1, 2, 3])

rest_array = np.setdiff1d(total_array, sub_array)
rest_array

array([4, 5, 6, 7, 8, 9])