pandas - 如何计算列表数据的计数和第一次出现？

Question

我有一个包含汽车类型的 list ，
a = ['Car_1','Car_1','Car_1','Car_2','Car_3','Car_3']
我应该能够从上面的列表中创建 2 个列表的输出，

result_count = [1,0,0,1,1,0] #Whenever new car type is present in list, make it 1
result_count = [1,2,3,1,1,2] #Count each car type

如何在不使用 for 循环的情况下轻松实现这一目标？谢谢。

Answer 1

IIUC，使用 Pandas 数据框

对于 #1，我们可以使用 .drop_duplicates
对于#2，我们可以使用 groupby和 cumcount

df = pd.DataFrame({'cars' : a})

one = (~df['cars'].duplicated()).astype(int).tolist() # Thanks to Erfan
two = (df.groupby('cars').cumcount() + 1).tolist()

print(one)

[1 0 0 1 1 0]

print(two)

[1, 2, 3, 1, 1, 2]