如何在Pandas身上加入df?

2024-03-29 10:21:23 发布

您现在位置:Python中文网/ 问答频道 /正文

如何加入我的数据帧?你知道吗

d = {'users': [1,2,3,4,5,6,7,8,9], 'code': [12,21,34,12,21,40,74,40,92]} 
df = pd.DataFrame(data=d)

我想用“代码”匹配我的“用户”。你知道吗

我要我的桌子像这样-

user1 user2 code active  
1      4      12   1
2      5      21   1
3     NONE    34   0
6      8      40   1

以此类推。。你知道吗


Tags: 数据代码用户nonedataframedfdatacode
1条回答
网友
1楼 · 发布于 2024-03-29 10:21:23

在我看来,最好创建包含id1、id2、代码和状态的类“Pair”。并存储这些对象的列表。你知道吗

但是如果您必须使用您的结构,您可以使用自定义函数来创建这样的数据帧。你知道吗

    def get_data_frame(data):
        columns = ['user1', 'user2', 'code', 'active']
        main_df = pd.DataFrame(columns=columns)
        for index, user in enumerate(data['users']):
            if data['code'][index] in main_df['code'].unique():
                main_df.loc[main_df['code'] == data['code'][index], ['user2']] = user
                main_df.loc[main_df['code'] == data['code'][index], ['active']] = True
            else:
                temp_df = pd.DataFrame(data=[[user, None, data['code'][index], False]], columns=columns)
                main_df = main_df.append(temp_df,  ignore_index=True)

        return main_df

相关问题 更多 >