Python Pandas：将列值设置为另一列的出现次数

Question

我的数据框两列上的 for 循环出现问题。 有了这个 for 循环，我想添加一个新列，正如我在这个例子中展示的那样：

我从这个数据框开始：

     person       store_id
0    Rose         24256
1    Tim          24257
2    Tim          27896
3    Tim          29547
4    Jack         24258
5    Jack         27897

我想添加的新列应该是这样的。 换句话说，我想添加一列显示每个人访问的商店的枚举。

     person       store_id      enum
0    Rose         24256         1
1    Tim          24257         1
2    Tim          27896         2
3    Tim          29547         3
4    Jack         24258         1
5    Jack         27897         2

这是我尝试过的代码，但没有得到我想要的。

for x in df['person']:
    i = 1
    for x in df['store_id']:
        df['enum'] = i
        i = i + 1

我也尝试了如下，再次得到错误的结果：

for x,y in zip(df['person'], df['store_id']):
    i = 1
    df['enum'] = i
    i = i + 1

我做错了什么？ 感谢您的帮助！

Answer 1

无需遍历多个列；使用iterrows 进行迭代并使用字典来跟踪访问过的商店 id 位置“enum”，然后设置 enum 列：

import pandas as pd
import numpy as np

df = pd.DataFrame(data={"person":[ 'Rose', 'Tim', 'Tim', 'Tim', 'Jack', 'Jack' ],
                    "store_id": [1580407549, 1580557630, 1580571344, 1580647900, 1580755371, 1580911127]})

store_index = {}
enums = []
for index, row in df.iterrows():
    sid = row['person']
    if sid not in store_index:
        store_index[sid] = 0
    store_index[sid] += 1
    enums.append(store_index[sid])
df['enum'] = enums
print(df)

输出：

  person    store_id  enum
0   Rose  1580407549     1
1    Tim  1580557630     1
2    Tim  1580571344     2
3    Tim  1580647900     3
4   Jack  1580755371     1
5   Jack  1580911127     2