지방이의 Data Science Lab

[python] 유저아이디별로 5줄씩만 가져오기 본문

Data Analysis/Python

[python] 유저아이디별로 5줄씩만 가져오기

[지현] 2019. 7. 23. 11:35

가지고 있는 데이터 프레임에서, 정렬해서 유저별로 상위 5줄씩가져와서 상위 아이템 보려고 할때:

top5 = {}

for num, idx in enumerate((pd.unique(lim.index))):

    print(num/len((pd.unique(kang.index))))

    top5[idx] = lim.loc[lambda x: x.index ==idx,'CEM_PRDCD'][0:5]

 

top5 = pd.concat(top5.values())

(굵은 표시 부분은 없어도 무방하다. %로 얼마나 진행됐는지 보고싶었는데, tqdm이 서버에 안깔려있어서 사용 불가했음)

 

out_dict = {}

for key in (pd.unique(top5.index))[0:100]:

    out_dict[key] = list((pd.DataFrame(top5).loc[lambda x: x.index == key, 'CEM_PRDCD']

 

pd.DataFrame..from_dict(out_dict, orient='index')

 

Comments