不妨以AB两种液体的消耗量为基础,我现在需要求出总的液体量。那么基本思路就是利用loc函数进行操作。对于loc函数,可以读取行和列,如果用冒号“:”则表示取全部的行或者全部的列。具体代码如下
import pandas as pd
df = pd.DataFrame({'液体A': [1, 2, 3], '液体B': [4, 5, 6]})
temp_data1 = df.loc[:,'液体B']
temp_data2 = df.loc[:,'液体A']
new = []
for i in range(0,len(temp_data1)):
temp = temp_data1[i] + temp_data2[i]
new.append(temp)
df.insert(loc=2, column='累计液体总用量', value=new)
print(df)