import pandas as pd df = pd.DataFrame([["A",1,2], ["B",3,4], ["A",5,6], ["C",7,8], ["A",9,10]], columns = ["colA", "colB", "colC"])