检查高维数组的有效方法在Python中的两个ndarray中重叠

发布时间：2020-12-20 12:10:00 所属栏目：Python 来源：网络整理

导读：例如,我有两个ndarray,train_dataset的形状是(10000,28,28),val_dateset的形状是(2000,28). 除了使用迭代之外,是否有任何有效的方法来使用numpy数组函数来查找两个ndarray之间的重叠？解决方法内存允许你可以使用 broadcasting ,就像这样 – val_dateset[(

例如,我有两个ndarray,train_dataset的形状是(10000,28,28),val_dateset的形状是(2000,28).

除了使用迭代之外,是否有任何有效的方法来使用numpy数组函数来查找两个ndarray之间的重叠？

解决方法

内存允许你可以使用 broadcasting,就像这样 –

val_dateset[(train_dataset[:,None] == val_dateset).all(axis=(2,3)).any(0)]

样品运行 –

In [55]: train_dataset
Out[55]: 
array([[[1,1],[1,1]],[[1,0],[0,0]],[[0,0]]])

In [56]: val_dateset
Out[56]: 
array([[[0,1]]])

In [57]: val_dateset[(train_dataset[:,3)).any(0)]
Out[57]: 
array([[[1,1]]])

如果元素是整数,则可以将输入数组中轴=(1,2)的每个块折叠为标量,假设它们是线性索引数,然后有效地使用np.in1d或np.intersect1d来查找匹配项.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!