numpy-如何方便地比較自定義dtype？

Question

我有一個超過30個字段的dtype。 我想比較具有該dtype的兩個對象，以便我確切地知道哪些字段不相等。 一個簡單的解決方案是在一系列if語句中對每個字段比較進行硬編碼：

if (obj1['field1']==obj2['field1']) DO_SOMETHING
if (obj1['field2']==obj2['field2']) DO_SOMETHING
# ...  

有沒有更好的方法比較兩個具有自定義dtypes的對象並確切知道哪些字段匹配或不匹配？

Answer 1

您可以通過OBJECT.dtype.names訪問對象的OBJECT.dtype.names字段。 所以：

# obj1 and obj2 are elements in a numpy array with a custom dtype
for field in obj1.dtype.names:
    if obj1[field]==obj2[field]:
        # DO_SOMETHING