转换numpy矩阵数据

Question

我有一个python的2D矩阵，有8列。 这8列保存了两个不同实体的记录，比如A和B，每个实体有4列。我的示例数据如下：

╔══════╦═════════╦══════╦═════════╗
║ A_ID ║ arrtibA ║ B_ID ║ attribB ║
╠══════╬═════════╬══════╬═════════╣
║    1 ║ val     ║    1 ║ val     ║
║    1 ║ val     ║    1 ║ val     ║
║    1 ║ val     ║    2 ║ val     ║
║    1 ║ val     ║    2 ║ val     ║
║    2 ║ val     ║    3 ║ val     ║
║    2 ║ val     ║    3 ║ val     ║
║    2 ║ val     ║    4 ║ val     ║
║    3 ║ val     ║    5 ║ val     ║
║    3 ║ val     ║    5 ║ val     ║
╚══════╩═════════╩══════╩═════════╝

我想把它变成这样：

╔══════╦═════════╦══════╦═════════╗
║ A_ID ║ arrtibA ║ B_ID ║ attribB ║
╠══════╬═════════╬══════╬═════════╣
║    1 ║ val     ║    1 ║ val     ║
║    1 ║ val     ║    1 ║ val     ║
║    2 ║ val     ║    2 ║ val     ║
║    2 ║ val     ║    2 ║ val     ║
║    3 ║ val     ║    3 ║ val     ║
╚══════╩═════════╩══════╩═════════╝

简单地说，根据实体A和实体B各自的id来对齐它们的属性。找不到匹配的行可以丢弃。我有28000多行，其中A\u ID从17到230，B\u ID从17到305。但是一旦对齐完成，两个实体的id都在231以下的行将在彼此前面对齐，实体B的231到305之间的其余行将被丢弃。你知道吗

Answer 1

您可以尝试：

data = np.delete(data, np.where(data[:,0]!=data[:,2])[0], 0)