我有一个下面的数据框:
user | speed
------------
Anna | 1.0
Bell | 1.2
Anna | 1.3
Chad | 1.5
Bell | 1.4
Anna | 1.1
我想使用字典来记录每个用户的遇到次数,并在循环数据帧时更新他/她的速度。
例如,我们第一次看到“Anna”时,字典是:
{"Anna": [1, 1.0]}
当我们第二次看到“Anna”时,它变成:
{"Anna": [2, 1.3], "Bell": [1, 1.2]}
最终字典应该是:
{"Anna": [3, 1.1], "Bell": [2, 1.4], "Chad": [1, 1.5]}
计数部分很简单:
>>> import pandas as pd
>>> record = pd.DataFrame({"user": ("Anna", "Bell", "Anna", "Chad", "Bell", "Anna"), "speed": (1.0, 1.2, 1.3, 1.5, 1.4, 1.1)})
>>> record
speed user
0 1.0 Anna
1 1.2 Bell
2 1.3 Anna
3 1.5 Chad
4 1.4 Bell
5 1.1 Anna
>>> encounter = {}
>>> for i in record['user']:
... encounter[i] = encounter.get(i, 0) + 1
...
>>> encounter
{'Anna': 3, 'Bell': 2, 'Chad': 1}
但是创建列表的空字典并更新第二个值的好方法是什么?谢谢!
最佳答案
我相信这就是您想要的两行内容。
import pandas as pd
record = pd.DataFrame({
"user": ("Anna", "Bell", "Anna", "Chad", "Bell", "Anna"),
"speed": (1.0, 1.2, 1.3, 1.5, 1.4, 1.1)
})
encounter = {}
for name, value in zip(record["user"], record["speed"]):
encounter[name] = [encounter.get(name, [0])[0] + 1, value]
zip
方法可让您同时循环显示名称和速度。get
方法尝试获取记录(如果存在),否则返回列表[0]
。- 第二个
[0]
采用列表的第一个元素,即计数器。 - 最后它与新值结合,成为一个新列表,并分配给
encounter[name]
。
关于python - 在Python中初始化列表的字典并更新其值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51238873/