在熊猫中对有条件的行求和

时间:2020-08-03 16:10:34

标签: python pandas

我想读取一个名为“ data2”的文件,并为每一行计算“ 5s”之和,然后将它们写入文件。我的以下代码创建了一个名为“总计”的列并进行打印。我的问题是我的代码只是写回我的原始数据。

id     score1 score2 score3 score4
1425     5      3     5       5
6271     4      3     5       2
81729    2      3     3       3
515268   5      3     2       5

和代码:

import pandas as pd


df = pd.read_csv("data2",index_col=0)
df['total'] = df.iloc[:,1:-1].eq(5).sum(axis=1)

print(df['total'])

0 个答案:

没有答案