例如，我有一個熊貓數據框
| 第 1 列 | Col2 | 第 3 列 |
|---|---|---|
| -1 | 2 | -3 |
我想選擇最小絕對值： table2
| 第 1 列 | Col2 | 第 3 列 | 結果 |
|---|---|---|---|
| -1 | 2 | -3 | -1 |
現在我正在使用df.abs().idxmin(axis="columns")，我得到：table3
| 第 1 列 | Col2 | 第 3 列 | 結果 |
|---|---|---|---|
| -1 | 2 | -3 | 第 1 列 |
我想問一下如何將表3轉換為表2？
使用np.argmin（numpy 對應物DataFrame.idxmin）。由於您要提取原始值，因此在 numpy 級別訪問這些值會更方便。
我在您的 MRE 中添加了一行以進行演示：
cols = np.argmin(df.abs().to_numpy(), axis=1) # [0, 2]
rows = range(len(cols)) # [0, 1]
df['Result'] = df.to_numpy()[rows, cols]
# Col1 Col2 Col3 Result
# 0 -1 2 -3 -1
# 1 10 -5 4 4
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句