我有这样一个数据帧:
| idx | Var1 | Var2 | Var3 |
|---|---|---|---|
| 0 | True | False | False |
| 1 | False | True | False |
| 2 | True | False | True |
| 3 | False | False | False |
| 4 | True | False | True |
我想创建三个新列，其中距离(每行)最接近True，如果该行有True show 0，那么我将得到以下结果:
| idx | Var1 | Var2 | Var3 | distV1 | distV2 | distV3 |
|---|---|---|---|---|---|---|
| 0 | True | False | False | 0 | 1 | 2 |
| 1 | False | True | False | 1 | 0 | 1 |
| 2 | True | False | True | 0 | 1 | 0 |
| 3 | False | False | False | 1 | 2 | 1 |
| 4 | True | False | True | 0 | 3 | 0 |
我已经阅读了与这个话题相关的所有其他讨论，但没有找到类似问题的答案.
以下是numpy ops的一种方法:
for c in df:
r = np.where(df[c])[0]
d = abs(df.index.values[:, None] - r)
df[f'{c}_dist'] = abs(df.index - r[d.argmin(1)])
print(df)
Var1 Var2 Var3 Var1_dist Var2_dist Var3_dist
0 True False False 0 1 2
1 False True False 1 0 1
2 True False True 0 1 0
3 False False False 1 2 1
4 True False True 0 3 0