合并行中的值取决于列中的值

时间:2015-03-16 05:14:51

标签: python

让我说我有

col1  col2
Aa    1
As    2
Aa    1
Aj    4
af    2
As    3

程序

for rows in data_set:
      rows.strip()
      rows.split('\t')
      if rows[0] # this is where I'm lost

如果Col1中的值彼此相等,则合并它们的值,以便存在一行添加了Col2值。例如,这里的第一行是

Col1    Col2
Aa      2
As      5

依此类推......

1 个答案:

答案 0 :(得分:1)

您可以使用dict存储Col2值。

dict = {}
for rows in data_set:
      rows.strip()
      rows.split('\t')
      if rows[0] in dict:
         dict[rows[0]] += rows[1]
      else:
         dict[rows[0]]  = rows[1]

然后你的dict看起来像这样

{'Aa':2,'As':5}