我有以下的pandas DataFrame。
import pandas as pd
df = pd.read_csv('filename.csv')
print(df)
dog A B C
0 dog1 0.787575 0.159330 0.053095
1 dog10 0.770698 0.169487 0.059815
2 dog11 0.792689 0.152043 0.055268
3 dog12 0.785066 0.160361 0.054573
4 dog13 0.795455 0.150464 0.054081
5 dog14 0.794873 0.150700 0.054426
.. ....
8 dog19 0.811585 0.140207 0.048208
9 dog2 0.797202 0.152033 0.050765
10 dog20 0.801607 0.145137 0.053256
11 dog21 0.792689 0.152043 0.055268
....
我通过以下方式将列"A"
,"B"
和"C"
相加以创建一个新列:
df['total_ABC'] = df[["A", "B", "B"]].sum(axis=1)
现在我想基于一个条件来执行这个操作,即如果 "A" < 0.78
,则创建一个新的求和列 df['smallA_sum'] = df[["A", "B", "B"]].sum(axis=1)
。否则,该值应为零。
如何创建类似这样的条件语句?
我的想法是使用
df['smallA_sum'] = df1.apply(lambda row: (row['A']+row['B']+row['C']) if row['A'] < 0.78))
然而,这并不起作用,我无法指定轴。
如何根据其他列的值创建列?
您也可以执行类似于每个df ['dog'] =='dog2'
,创建列dog2_sum
的操作,即:
df['dog2_sum'] = df1.apply(lambda row: (row['A']+row['B']+row['C']) if df['dog'] == 'dog2'))
但是我的做法不正确。