根据空值将计算列添加到pandas DataFrame,可以通过以下步骤实现:
import pandas as pd
import numpy as np
data = {'A': [1, 2, np.nan, 4, 5],
'B': [np.nan, 2, 3, np.nan, 5],
'C': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)
fillna()
函数将DataFrame中的空值填充为特定的值,例如0:df_filled = df.fillna(0)
这将把DataFrame中的所有空值替换为0。
apply()
函数和lambda
表达式来添加计算列。假设我们想要将列A和列B相加,并将结果存储在新的列D中:df_filled['D'] = df_filled.apply(lambda row: row['A'] + row['B'], axis=1)
这将对DataFrame中的每一行执行lambda表达式,将列A和列B相加,并将结果存储在列D中。
print(df_filled)
完整的代码示例如下:
import pandas as pd
import numpy as np
data = {'A': [1, 2, np.nan, 4, 5],
'B': [np.nan, 2, 3, np.nan, 5],
'C': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)
df_filled = df.fillna(0)
df_filled['D'] = df_filled.apply(lambda row: row['A'] + row['B'], axis=1)
print(df_filled)
这样,我们就根据空值将计算列添加到了pandas DataFrame。在实际应用中,可以根据具体需求进行更复杂的计算和处理。
领取专属 10元无门槛券
手把手带您无忧上云