numpy数组,如果在另一行中重复一个字符串元素,则删除行

时间:2015-11-22 21:19:52

标签: python arrays

我有一个这样的数组:

vals = numpy.array([['user1','number1','grades1'],
                    ['user1', 'number2','grade2'], 
                    ['user3', 'number3','grade3'], 
                    ['user4', 'number4','grade4']])

我想删除包含user1的整行,因为在第一列的另一行中重复(无论哪一行无关)。此外,如果有更多行显示user1,请保留找到的第一行。所以输出可以是这样的:

array([['user1', 'number1','grade1'], 
       ['user3', 'number3','grade3'], 
       ['user4', 'number4','grade4']])

到目前为止,我试过了:

a = (vals[0:1,0:1] == vals[1:2,0:1])
vals = numpy.delete(vals, numpy.where(a), axis=0)

但是,如何使用这个制作循环?

0 个答案:

没有答案