如何用单独的dict值替换dataframe列 - python

问题描述 投票:0回答:1

我下面的user_artist_plays数据框显示了一个用户列,但是对于统计计算,我必须用int只用ID替换这些混合字符。

    users                                       artist  plays
0   00001411dc427966b17297bf4d69e7e193135d89    sting   12763
1   00001411dc427966b17297bf4d69e7e193135d89    stars   8192
2   fffe8c7f952d9b960a56ed4dcb40a415d924b224    cher    117
3   fffe8c7f952d9b960a56ed4dcb40a415d924b224    queen   117

上面只显示了两个用户的多个条目,如果我可以让列匹配单独字典中现有密钥的任何条目,那就没问题:

users = user_artist_plays['users'].unique()
user_dict = {ni: indi for indi, ni in enumerate(set(users))}
user_dict

{'068156fafd9c4237c174c648d3d484cbf509cb75': 0,
 '6deecfbc46a81e4faf398b2afd991be05ab78f10': 74205,
 '1e23333ff4f637420a8a38d467ccecfda064afb9': 1,
 '0b282cafc949efe4163b7946b7104957a18cf010': 2,
 'd1867cbda35e0d48e9a8390d9f5e079c9d99ea96': 3}

这是我尝试切换int值:

for k, v in user_dict.items():
        if user_artist_plays['users'].any(k):
            user_artist_plays['users'].replace(v)

它保留了users列的原始值......

python pandas dataframe dictionary-comprehension
1个回答
3
投票

看来你需要map

user_artist_plays['users'] = user_artist_plays['users'].map(user_dict)

或者factorize

user_artist_plays['users'] = pd.factorize(user_artist_plays['users'])[0]
© www.soinside.com 2019 - 2024. All rights reserved.