I'd like to have the indexes of duplicated column elements as a list. So far, the way I found is
test = ['a', 'a', 'b', 'c', 'b']
testdf = pd.DataFrame(test, columns=['test'])
np.asarray(np.where(list(testdf['test'].duplicated()))).tolist()[0]
# [1, 4]
Which seems ridiculously convoluted.
Any better way?
CodePudding user response:
you can use .duplicated()
with .tolist()
testdf.index[testdf.test.duplicated()].tolist()
CodePudding user response:
Try this just indexing the index:
testdf.index[testdf['test'].duplicated()]
add to_list
:
testdf.index[testdf['test'].duplicated()].to_list()
Output:
[1, 4]