pandas - 获取两列之间的最小值并分配如果两列不是 null

Question

如果两列都不是 null，我试图弄清楚如何在两列之间分配最小值。 因此，给定一个 dataframe 并连续填充以下数据：

col1    col2    col3
347     933     338
938     523     211

我正在尝试将临时列分配给 col2 和 col3 之间的最小值，但以下给了我一个错误：

df.loc[df['col2'].notnull() & df['col3'].notnull(), 'TEMP_rank'] = min(df.loc[['col2'], df['col3']]).min().min()

我也有以下问题：

df.loc[df['col2'].notnull() & df['col3'].notnull(), 'TEMP_rank'] = min(df.loc[['col2'], df['col3']]).min(1)

我正在寻找以下 output （在第 2 列和第 3 列之间进行测试）：

col1    col2    col3    tempCol
347     933     338     338
938     123     211     123

Answer 1

如果您只想在 null / NaN 都不是时计算min() ，则可以这样做。

df = pd.read_csv(io.StringIO("""col1    col2    col3
347     933     338
500     NaN     200
938     523     211"""), sep="\s+")

df = df.assign(
    tempCol=lambda dfa: np.where(dfa["col2"].isna()|dfa["col3"].isna(), 
                                 np.nan, 
                                 dfa.loc[:,["col2","col3"]].min(axis=1))
)

output

   col1   col2  col3  tempCol
0   347  933.0   338    338.0
1   500    NaN   200      NaN
2   938  523.0   211    211.0