Python (3.7) CSV 按字段值排序/求和

Question

我有一个 csv 文件(大小不定)，我想阅读该文件并对其进行一些处理。以下是 csv 文件的结构:

User, Value
CN,500.00
CN,-250.00
CN,360.00
PT,200.00
PT,230.00
...

我想读取文件并获取第一个字段相同的每一行的总和。我一直在尝试以下操作，只是为了尝试确定第一个字段的值:

with open("Data.csv", newline='') as data:
    reader = csv.reader(data)
    for row in reader:
        if row.startswith('CN'):
            print("heres one")

此操作失败，因为 startswith 不适用于列表对象。我也尝试过使用 readlines()。

编辑1:

我目前可以打印以下带有排序总和的数据框对象:

         Value
User
CN    3587881.89
D        1000.00
KC    1767783.99
REC     12000.00
SB      25000.00
SC    1443039.12
SS          0.00
T     9966998.93
TH    2640009.32
ls        500.00

我使用以下代码得到此输出:

mydata=pd.read_csv('Data.csv')
out = mydata.groupby(['user']).sum()
print(out)

我现在希望能够为此对象编写 if 语句。像这样的东西:

if out contains User 'CN'
    varX = Value for 'CN'

因为现在这是一个数据帧类型，所以我无法将值设置为特定用户的变量。

Answer 1

您可以执行以下操作:

import pandas as pd
my_data= pd.read_csv('Data.csv')
my_data.group_by('user').sum()