python - Pandas 使用 while 循环遍历数据框和列表

Question

我试图循环遍历一个列表和一个数据帧，如果列表中的 id 等于数据帧中的 id，则对数据帧中的该行执行某些操作。

import pandas as pd
data = [['a1','Alex',10],['a1','Bob',12],['a1','Clarke',13],['a2','den',14],['a2','emry',15]]
df = pd.DataFrame(data,columns=['id','Name','Age'])

unique_ids = ['a1','a2']

首先循环遍历列表。如果数据框中的 id == unique_ids 列表中的 id，则执行以下操作:

• 如果下一行中的唯一 ID 仍然与前一行相同，则将第二个参数设置为上一行中的最后一个值。因此，由于 12 是第一行中的最后一项，并且 a1 仍然是与上面相同的 id，因此将 12 设置为第二行中的第二个值。

例如:上述输入的预期输出为

a1,10,12 
a1,12,13 
a2,14,15

我是如何尝试的:

for i in unique_ids:
    for row in df.itertuples(index=True, name='Pandas'):
        while i == getattr(row,"id"):
           print (getattr(row,"id"),getattr(row,"age")
           not sure how to proceed as im getting stuck at the while loop

Answer 1

我认为你想要做的事情可以通过跟踪最后一行的 id 来完成。

import pandas as pd
data = [['a1','Alex',10],['a1','Bob',12],['a1','Clarke',13],['a2','den',14],['a2','emry',15]]
df = pd.DataFrame(data,columns=['id','Name','Age'])

unique_ids = ['a1','a2']
last_id = df.iloc[0]['id']  # initilize to the first row's id
for idx, row in df[1:].iterrows():  
    if row['id'] in unique_ids and row['id'] == last_id:
        # You can retrieve last row by df.iloc[idx-1]
        print(row['id'], ",", df.iloc[idx-1]['Age'], ",", row['Age']) 
    last_id = row['id'] # update last_id

Output:
a1 , 10 , 12
a1 , 12 , 13
a2 , 14 , 15