Python Pandas Dataframe-在A列中的每个标签上查找B列中的和
假设我们有以下数据:Python Pandas Dataframe-在A列中的每个标签上查找B列中的和,python,pandas,multiple-entries,Python,Pandas,Multiple Entries,假设我们有以下数据: ... col1 col2 col3 0 A 1 info 1 A 2 other 2 B 3 blabla 我想使用python查找重复的条目(在第1列中),并根据第2列将它们相加 在python中,我将执行以下操作: l = [('A',1), ('A',2), ('B',3)] d = {} for i in l: if(i[0] not
... col1 col2 col3
0 A 1 info
1 A 2 other
2 B 3 blabla
我想使用python查找重复的条目(在第1列中),并根据第2列将它们相加
在python中,我将执行以下操作:
l = [('A',1), ('A',2), ('B',3)]
d = {}
for i in l:
if(i[0] not in d.keys()):
d[i[0]]=i[1]
else:
d[i[0]]=d[i[0]]+i[1]
print(d)
因此,结果将是:
{'A': 3, 'B': 3}
使用熊猫做同样的事情有没有简单的方法?使用:
这工作正常,但是如果我想保留第1列呢?在这种情况下,第1列中的属性与索引一样使用。
In [1]: import pandas
In [2]: df = pandas.DataFrame({"col1":["A", "A", "B"], "col2":[1,2,3]})
In [3]: df.groupby("col1").sum()
Out[3]:
col2
col1
A 3
B 3
In [4]: df.groupby("col1").sum().reset_index()
Out[4]:
col1 col2
0 A 3
1 B 3
[2 rows x 2 columns]