3

我有三列数据报，打算以给定的格式在字典中转换：

数据报：

user_id item_id ratings
3         2       3
3         3       4
1         3       1
2         1       4

用户数 = 3

项目编号 = 3

评级 = 0 到 5

字典=

{user_id1:[rating_for_item1, rating_for_item2, rating_for_item3],
 user_id2:[.same as previous.],
 user_id3:[..same as prev..]}

例如，

{1:[0,0,1], 2:[4,0,0], 3:[0,3,4]}

所以，我能做的就是输出如下：

{1:{3:1}, 2:{1:4}, 3:{2:3, 3:4}} #{user_id:{item_id:rating}.....}

上面输出的代码如下：

import pandas as pd
data = {}
cols = ['user_id', 'item_id', 'ratings']
pf = pd.read_csv('filename', sep='\t', names= cols)
for user, item, rate in pf.values: data.setdefault(user,{})[item] = rate
print data

我的代码中缺少什么，或者我完全走错了路。请帮忙。

4

2 回答 2

2

我会pivot然后建立字典。例如：

pdf = df.pivot("user_id", "item_id").fillna(0)
d = {k: v.tolist() for k,v in pdf.iterrows()}

生产

>>> d
{1: [0.0, 0.0, 1.0], 2: [4.0, 0.0, 0.0], 3: [0.0, 3.0, 4.0]}

一、框架：

>>> df
   user_id  item_id  ratings
0        3        2        3
1        3        3        4
2        1        3        1
3        2        1        4

枢：

>>> pdf = df.pivot("user_id", "item_id")
>>> pdf
         ratings        
item_id        1   2   3
user_id                 
1            NaN NaN   1
2              4 NaN NaN
3            NaN   3   4

将NaNs 替换为 0：

>>> pdf = df.pivot("user_id", "item_id").fillna(0)
>>> pdf
         ratings      
item_id        1  2  3
user_id               
1              0  0  1
2              4  0  0
3              0  3  4

并使用字典理解构建逐行字典：

>>> d = {k: v.tolist() for k,v in pdf.iterrows()}
>>> d
{1: [0.0, 0.0, 1.0], 2: [4.0, 0.0, 0.0], 3: [0.0, 3.0, 4.0]}

有很多方法可以完成这最后一步，包括dict(zip(pdf.index, pdf.values.tolist()))，但是当您想稍微调整一下时，它们中的许多方法并不容易概括。

于 2013-10-31T18:21:04.903 回答
1

如何将您拥有的东西处理成您想要的东西：

from collections import defaultdict

processed_data = defaultdict(list)
for k,v in data.items():
    for idx in range(1, 4): # Make sure we check each item
                            # from (1 to 3 inclusive) for each iteration
                            # of the dictionary
        val = v.get(idx, 0)
        processed_data[k].append(val)

processed_data产量：

defaultdict(<type 'list'>, {1: [0, 0, 1], 2: [4, 0, 0], 3: [0, 3, 4]})

如果您想将其转换回常规字典（从defaultdict,），请执行以下操作：

dict(processed_data)

产生

{1: [0, 0, 1], 2: [4, 0, 0], 3: [0, 3, 4]}
于 2013-10-31T17:56:42.870 回答